Codeforces Round #621 (Div. 1 + Div. 2) D. Cow and Fields （最短路SPFA）

题目链接


题意：给定一个n点m边的无向图，现在有k个点，要求你在这k个点里选两个点给他们连边，求连边后，最短路的最大值是多少。
思路：由于我们并不知道将两个点连边之后会不会对初始的最短路造成影响，所有我们可以先求两边spfa，分别求以1为起点和以n为起点的各点最短路，我们要是最短路损失的尽量小，肯定是这k个点尽可能相邻，所以求第一遍以1为起点的spfa之后，可以对这k个点进行排序，每次比较相邻两个点对答案的影响。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+1;
const int inf=0x3f3f3f3f;
bool vis[maxn];
int n,m,k,ans,a[maxn],d1[maxn],d2[maxn];
vector<int>g[maxn];
vector<pair<int,int>>temp;
bool cmp(const pair<int,int>&a,const pair<int,int>&b)
{
	return a.second<b.second;
}
void spfa(int d[],int x)
{
	memset(vis,false,sizeof(vis));
	fill(d,d+maxn,inf);
	d[x]=0;
	queue<int>q;
	q.push(x);
	vis[x]=true;
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		vis[top]=false;
		for(auto v:g[top])
		{
			if(d[top]+1<d[v])
			{
				d[v]=d[top]+1;
				if(!vis[v]) q.push(v),vis[v]=true;
			}
		}
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=k;++i) scanf("%d",&a[i]);
	for(int i=1;i<=m;++i){
		int u,v;
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	spfa(d1,1);
	spfa(d2,n);
	for(int i=1;i<=k;++i) temp.push_back({a[i],d1[a[i]]});
	sort(temp.begin(),temp.end(),cmp);
	for(int i=1;i<temp.size();++i)
	{
		int t=d1[n],x=temp[i-1].first,y=temp[i].first;
		t=min(t,min(d1[x]+d2[y]+1,d1[y]+d2[x]+1));
		ans=max(ans,t);
	}
	printf("%d\n",ans);
 }