团体程序设计天梯赛L2-001 紧急救援

时间:2023-02-13 11:58:38

L2-001. 紧急救援

时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越

作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。

输入格式:

输入第一行给出4个正整数N、M、S、D,其中N(2<=N<=500)是城市的个数,顺便假设城市的编号为0~(N-1);M是快速道路的条数;S是出发地的城市编号;D是目的地的城市编号。第二行给出N个正整数,其中第i个数是第i个城市的救援队的数目,数字间以空格分隔。随后的M行中,每行给出一条快速道路的信息,分别是:城市1、城市2、快速道路的长度,中间用空格分开,数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式:

第一行输出不同的最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔,输出首尾不能有多余空格。

输入样例:
4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2
输出样例:
2 60
0 1 3

——————————————————————————————————

题目的意思是求从S到D的最短路,并输出最短路的条数和点权和最大的路

在Dijkstra时开几个数组更新即可,由于求条数,更新时等于的情况特殊处理

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <stack>
#include <string>
#include <set>
#include<vector>
#include <map>
using namespace std;
#define inf 0x3f3f3f3f
int edge[505][505];
int vis[505],cnt[505],dis[505],sum[505],pre[505];
int a[505];
struct node{
	int id,val;
	bool friend operator<(node a,node b)
	{
		return a.val>b.val;
	}
};

int n,m;

void djstl(int o)
{
	memset(dis,inf,sizeof dis);
	memset(vis,0,sizeof vis);
	memset(sum,0,sizeof sum);
	memset(cnt,0,sizeof cnt);
	node f,d;
	f.id=o;
	f.val=0;
	priority_queue<node>q;
	q.push(f);
	vis[o]=1;
	dis[o]=0;
	sum[o]=1;
	cnt[o]=a[o];
	while(!q.empty())
	{
		f=q.top();
		q.pop();
		vis[f.id]=1;

		for(int i=0;i<n;i++)
		{
			if(!vis[i]&&f.val+edge[f.id][i]<dis[i])
			{
				dis[i]=f.val+edge[f.id][i];
				cnt[i]=cnt[f.id]+a[i];
				pre[i]=f.id;
				sum[i]=sum[f.id];
				d.id=i;
				d.val=dis[i];
				q.push(d);
			}
			else if(!vis[i]&&f.val+edge[f.id][i]==dis[i])
			{
				if(cnt[i]<cnt[f.id]+a[i])
				{
					cnt[i]=cnt[f.id]+a[i];
					pre[i]=f.id;
				}
				sum[i]+=sum[f.id];

			}
		}
	}
}


int main()
{
	int st,ed,u,v,w;
		while(~scanf("%d%d%d%d",&n,&m,&st,&ed))
		{
			for(int i=0;i<n;i++)
				scanf("%d",&a[i]);
			memset(edge,inf,sizeof edge);
			for(int i=0;i<m;i++)
			{
				scanf("%d%d%d",&u,&v,&w);
				if(w<edge[u][v])
				{
				edge[u][v]=edge[v][u]=w;
				}
			}
			memset(pre,-1,sizeof pre);
			djstl(st);
			printf("%d %d\n",sum[ed],cnt[ed]);
			stack<int>s;
			s.push(ed);
			while(pre[ed]!=-1)
            {
            s.push(pre[ed]);
            ed=pre[ed];
            }
            int q=0;
            while(!s.empty())
            {
                if(q++)
                    printf(" ");
                printf("%d",s.top());
                s.pop();
            }
            printf("\n");
		}
		return 0;
}