题目链接: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;
}