L2-001 紧急救援 (25 分)

 

输入样例:

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

解题思路：

本题需要维护两个变量 路径最短前提下救援人数最多，还要求出经过的点数和路径。

路径最短用dis去维护这个不用多说，救援人数最多用w[i]来维护，更新方式写在最短路的for循环里，具体的赋值和更新于dis基本上相同，点数也是如次，用sum[i]进行维护。

路径这里使用了并查集的思想，只需要记住每个节点的父节点，依次向前搜索即可得到最终路径。

心得：这道题整整捣鼓了两个小时，分享一下心得。

1.发现一种新的更新方式，把地图全部更新为inf，i==j时不用更新0,dis全部更新为inf,dis[1]=0，q其他和正常步骤一样去求最短路也能得到最终答案，并且这样求是从起始点开始遍历，这样求路径的时候不会漏点。

2.如果最短路上需要增加其他权值（比如本题中需要维护路径最短得前提下维护救援人数最大），更新方式和dis得更新方式完全一样，先全部更新为0，初始点更新为这个点得权值，把维护得步骤写在最后更新最短路得for循环中即可。

3.本题中在存路径时使用的是并查集得思想。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<map>
#include<vector>
#include<set>
#include<cmath>
#include<queue>
#define int long long
#define endl "\n"

using namespace std;

const int inf=0x3f3f3f3f;

int n,m,s,d,sum[550];
int mp[550][550],wig[550],dis[550],book[550];
int f[550],w[550],way[550],k;

void find(int x)
{
	if(x==s) 
	{
		cout<<x-1;
		return ;
	}
	find(f[x]);
	cout<<" "<<x-1;
}


void dij()
{
	memset(book,0,sizeof book);
	memset(w,0,sizeof w);
	memset(sum,0,sizeof sum);
	for(int i=1;i<=n;i++) dis[i]=inf;
	dis[s]=0;
	w[s]=wig[s];
	sum[s]=1;
	for(int i=1;i<=n;i++)
	{
		int mi=inf,u=0;
		for(int j=1;j<=n;j++)
		{
			if(!book[j] && mi>dis[j])
			{
				mi=dis[j],u=j;
			}
		}
		book[u]=1;
		for(int v=1;v<=n;v++)
		{
			if(!book[v])
			{
				if(dis[v]>dis[u]+mp[u][v])
				{
					dis[v]=dis[u]+mp[u][v];
					w[v]=w[u]+wig[v];
					sum[v]=sum[u];
					f[v]=u;
					
				}
				else if(dis[v]==dis[u]+mp[u][v])
				{
					if(w[v]<w[u]+wig[v])
					{
						w[v]=w[u]+wig[v];
						f[v]=u;
					}
					sum[v]+=sum[u];	
				}
			}
		}
	}
}

void solve()
{
	cin>>n>>m>>s>>d;
	s++,d++;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			mp[i][j]=inf;
	
	for(int i=1;i<=n;i++) cin>>wig[i];
	for(int i=1;i<=m;i++)
	{
		int a,b,c;cin>>a>>b>>c;a++,b++;
		if(mp[a][b]>c) mp[a][b]=mp[b][a]=c;
	}
	
	dij();
	cout<<sum[d]<<" "<<w[d]<<endl;
	find(d);
	
	
}

signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int T=1;//cin>>T;
	while(T--) solve();
}