寒假刷题31: Codeforces Round #621 D. Cow and Fields

题目链接：

Codeforces Round #621 D. Cow and Fields

题目解析：

首先用BFS计算出起点1和终点n到各个点的距离,假设任意一点到起始点的距离为Xi,到终止点距离为Yi。
对于任意两个特殊点a,b,如果在他们之间修路，那么经由他们的距离为
min(Xa+Yb,Ya+Xb)
这样很容易想到一种O（n^2）的做法：枚举所有点，计算距离
如何优化呢？假设Xa+Yb<Ya+Xb,移项得Xa-Ya<Xb-Yb。因此我们可以按照Xi-Yi排序，之后从左到右遍历数组，对于每一个点i，答案一定是
max(在i之前所有点的X)+Yi+1
注意：有可能不经过特殊点的一条路径是最短路，因此最后答案要在之前的最短路和经过特殊点的最短路中取min

AC代码：

#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
#define ll long long
int n,m,k;
const int MAXN=2e5+5;
vector<int>g[MAXN];
int vis[MAXN],d[2][MAXN];//d[0][i]表示1到i的最短路距离,d[1][i]表示n到i的最短路距离
struct fields
{
    int num;
    int to1,ton;
}a[MAXN];
bool mycmp(fields x,fields y){ return x.to1-x.ton<y.to1-y.ton;}
void bfs(int st,int id)
{
    queue<int>q;
    q.push(st);
    for(int i=1;i<=n;i++)vis[i]=0;
    vis[st]=1;
    while(!q.empty())
    {

        int x=q.front();//cout<<x<<endl;
        q.pop();
        for(int i=0;i<g[x].size();i++)
        {
            int v=g[x][i];
            //cout<<v<<endl;
            if(vis[v])continue;
            d[id][v]=d[id][x]+1;
            q.push(v);
            vis[v]=1;
        }
    }
}
int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++) cin>>a[i].num;
    int x,y;
    for(int i=1;i<=m;i++)
    {
        cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    bfs(1,0);
    bfs(n,1);
    for(int i=1;i<=k;i++) a[i].to1=d[0][a[i].num],a[i].ton=d[1][a[i].num];
    sort(a+1,a+k+1,mycmp);
    int ans=-1,maxx=0;
    for(int i=1;i<=k;i++)
    {
        if(i>1) ans=max(ans,maxx+a[i].ton+1);
        maxx=max(maxx,a[i].to1);
    }
    cout<<min(ans,d[0][n])<<endl;
}