CodeForces 954D

题目链接：CF 954D

给定一个无向图和起点、终点

现在要向图中添加边，使得起点和终点之间的距离不会被缩小

两遍spfa处理出每个点分别距离起点和终点的最短的距离，双重循环判断如果dis（起点，i）+dis（j，终点）+1>dis（起点，终点），那么sum++；

因为点可能会倒过来所以要判断两次，详见代码

#include<iostream>
#include<cstdio>
#include<vector>
#include<set>
#include<map>
#include<string.h>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#define LL long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define sqr(a) (a)*(a)
#define lan(a,b) memset(a,b,sizeof(a))

using namespace std;

int tu[1010][1010];//邻接矩阵
int dis1[1010];//单源最短路径长度
int dis2[1010];
int n;//点数
bool spfa(int s,int dis[1010])
{
    queue<int> run;
    run.push(s);
    dis[s]=0;
    while(!run.empty())
    {
        int tem=run.front();
        run.pop();
        for(int i=1;i<=n;i++)
        {
            int p=tu[tem][i];
            if(p==inf)
                continue;
            if(dis[i]>dis[tem]+p)
            {
                run.push(i);
                dis[i]=dis[tem]+p;
            }
        }
    }
    return true;
}


int main()
{
    int m,s,t;
    while(~scanf("%d%d%d%d",&n,&m,&s,&t))
    {
        lan(tu,inf);
        lan(dis1,inf);
        lan(dis2,inf);
        for(int i=1;i<=m;i++)
        {
            int p,q;
            scanf("%d%d",&p,&q);
            tu[p][q]=tu[q][p]=1;
        }
        spfa(s,dis1);
        spfa(t,dis2);
        int sum=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n&&j!=i;j++)
        {
            if(i==t&&j==s)
                continue;
            if(dis1[i]+1+dis2[j]>=dis1[t]&&dis1[j]+1+dis2[i]>=dis1[t]&&tu[i][j]==inf)///判断两次
                sum++;
        }
        printf("%d\n",sum);
    }
    return 0;

}