【codeforces 954 D】 fight against traffic （最短路）

题意

一张无向连通图。

现在要求再加一条无向边，保证加边之后无重边自环，且 $s$ 到 $t$ 的最短路长度不变。

问有多少种加边方案。

思路

简单图论题，但是我胡了一个假的做法并且在长达一天的时间里一直以为自己是对的 qwq

做法是先分别求出所有点 $s$ 和 $t$ 的最短路。

然后枚举两个原图中没有连边的点 $i,j$，判断连上之后 $(s,i,j,t)$ 或者 $(s,j,i,t)$ 这两条路能否更新最短路，不能就计入答案。

然后来讲一下错误思路：对于 $s$ 求出所有点到他的距离。枚举 $s$ 到 $t$ 的最短路上的点 $i$， $dis[j]\le dis[i]-2$ 的点 $j$ 显然不能连，去掉。

但很显然这是错的，连接两个不在原图 $s$ 到 $t$ 最短路上的点是有可能更新最短路的。这个随便画个图就可以说明了。

代码

// cf 954 D
#include<bits/stdc++.h>
using namespace std;
const int N = 1000 + 10, M = 2000 + 10;
int n, m, s, t;
int h[N], ecnt, nxt[M], v[M];
bool mp[N][N];
int dis[2][N], ans;

void _add(int x, int y){
	nxt[++ecnt] = h[x]; v[ecnt] = y;
	h[x] = ecnt; mp[x][y] = 1;
}

queue<int> que;
void dijkstra(int s, int *dis){
	for (int i = 1; i <= n; ++ i) dis[i] = -1;
	que.push(s); dis[s] = 0;
	while (!que.empty()){
		int u = que.front(); que.pop();
		for (int i = h[u]; i; i = nxt[i])
			if (dis[v[i]] == -1){
				dis[v[i]] = dis[u]+1;
				que.push(v[i]);
			}
	}
}

int main()
{
	ios::sync_with_stdio(false);
	cin >> n >> m >> s >> t;
	ecnt = 1;
	for (int i = 1; i <= m; ++ i){
		int x, y;
		cin >> x >> y;
		_add(x, y); _add(y, x);
	}
	dijkstra(s, dis[0]);
	dijkstra(t, dis[1]);
	for (int i = 1; i <= n; ++ i)
		for (int j = i+1; j <= n; ++ j)
			if (!mp[i][j]){
				ans += min(dis[0][i]+dis[1][j]+1, dis[1][i]+dis[0][j]+1) >= dis[0][t];
			}
	cout << ans << endl;
	return 0;
}