Codeforces 954D Fight Against Traffic(BFS 最短路)

题目链接：Fight Against Traffic

题意：有n个点个m条双向边，现在给出两个点S和T并要增加一条边，问增加一条边且S和T之间距离不变短的情况有几种？

题解：首先dfs求一下S到其他点和T到其他点的最短路(好久不写有点手生@。@)，然后遍历所有的建边的情况，假设在i和j两个点之间建边则要满足 ds[i] + 1 + dt[j] > ds[T] && ds[j] + 1 + dt[i] > ds[T]。

#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 1e3+9;
const int INF = 1e9+9;
int N,M,S,T;
vector<int> dir[MAX_N];
int vec1[MAX_N],vec2[MAX_N];
void bfs(int pos , int* vec){
    fill(vec,vec+MAX_N,INF);
    queue<int> que;
    que.push(pos);
    vec[pos] = 0;
    while(!que.empty()){
        int t = que.front();
        que.pop();
        for(int i=0;i<dir[t].size();i++){
            int x = dir[t][i];
            if(vec[x] > vec[t] + 1){
                vec[x] = vec[t] + 1;
                que.push(x);
            }
        }
    }
    //for(int i=1;i<5;i++)cout<<"!!"<<vec[i]<<endl;
}
int main(){
    while(cin>>N>>M>>S>>T){
       for(int i=0;i<MAX_N;i++){
            dir[i].clear();
       }
       for(int i=0;i<M;i++){
           int a,b;
           scanf("%d%d",&a,&b);
           dir[a].push_back(b);
           dir[b].push_back(a);
       }
       bfs(S,vec1);
       bfs(T,vec2);
       int ans =0 ;
       int dis = vec1[T];
       for(int i=1;i<=N;i++){
            for(int j=i+1;j<=N;j++){
                if(vec1[i] + vec2[j] + 1 >= dis && vec1[j] + vec2[i] + 1 >= dis){
                   ans ++;
                }
            }
        }
        cout<<ans-M<<endl;
    }
    return 0;
}