【PTA】 L2-001 紧急救援 (25分) 【dijkstra+优先队列优化】

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

输入格式:
输入第一行给出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

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node{
    
    
	int v,w;
	node(int a,int b){
    
    v=a;w=b;}
	friend bool operator<(node a,node b){
    
    
		return a.w>b.w;
	}
};

const int maxn = 550;
const int inf = 1<<27; //表示无穷远,到不了
//p各地救援队,num路径数,dis最短距离,ans最大救援队,pre前驱节点
int n,m,s,d,a,b,len,p[maxn]={
    
    0},num[maxn]={
    
    0},dis[maxn],ans[maxn]={
    
    0},pre[maxn];
vector<node> g[maxn];
bool vis[maxn]={
    
    false}; //标记是否入过队

void print() //先放入栈中,再进行输出
{
    
    
	stack<int> s;
	s.push(d);
	while(pre[d]!=-1){
    
    
		s.push(pre[d]);
		d=pre[d];
	} 
	
	while(s.size())
	{
    
    
		cout<<s.top();
		s.pop();
		if(s.size()) cout<<" ";
	}
}

void dijk()
{
    
    
	fill(dis,dis+maxn,inf); //初始化距离为无穷远,表示到不了
	priority_queue<node> q;
	dis[s]=0; //起点距离起点0
	num[s]=1; //一条路径
	ans[s]=p[s]; //起点的最大救援队等于起点的救援队数
	pre[s]=-1; //设置起点的前驱节点为-1
	q.push(node(s,0));
	while(!q.empty())
	{
    
    
		int u=q.top().v;
		q.pop();
		if(vis[u]) continue;
		vis[u]=1; //标记入队
		for(int i=0;i<g[u].size();i++)
		{
    
    
			int v=g[u][i].v;
			int w=g[u][i].w;
			if(!vis[v]) //没入过队
			{
    
    
				if(dis[v]>dis[u]+w) //可以更新距离
				{
    
    
					pre[v]=u;
					dis[v]=dis[u]+w;
					ans[v]=ans[u]+p[v];
					num[v]=num[u];
				}
				else if(dis[v]==dis[u]+w) //距离相等
				{
    
    	
					if(ans[v]<ans[u]+p[v]) //可以更新最大救援队
					{
    
    
						pre[v]=u;
						ans[v]=ans[u]+p[v];
					}	
					num[v]+=num[u]; //累加路径条数
				}	
				q.push(node(v,dis[v]));
			}
		}
	}
}

int main()
{
    
    	   
	cin>>n>>m>>s>>d;
	for(int i=0;i<n;i++) cin>>p[i];
	for(int i=0;i<m;i++)
	{
    
    
		cin>>a>>b>>len;
		g[a].push_back(node(b,len)); //无向图
		g[b].push_back(node(a,len));
	}
	dijk();
	cout<<num[d]<<" "<<ans[d]<<endl;
	print();
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45260385/article/details/110247432