Codeforces 954D. Fight Against Traffic

原题链接
题意：
给出n点m边的无向图，现在要求在没有边的两点之前连边，并且s到t的最短路长度不变，问满足条件的边数。
思路：
从起点和终点分别跑一遍最短路，然后枚举一下未出现的边统计即可。

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
inline ll read()
{
    ll x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
ll ksm(ll a,ll b,ll p)
{
    ll res=1;
    while(b)
    {
        if(b&1)res=res*a%p;
        a=a*a%p;
        b>>=1;
    }
    return res;
}
#define PI acos(-1)
#define x first
#define y second
const int maxn=1e6+7,inf=0x3f3f3f3f;
int n,m,s,t;
struct node{
    int e,ne;
}edge[maxn];
int h[maxn],idx;
void add(int u,int v){
    edge[idx]={v,h[u]},h[u]=idx++;
}
int dis[maxn],st[maxn],dis1[maxn];
void dijkstra(int s){
    memset(dis,0x3f,sizeof dis);
    memset(st,0,sizeof st);
    dis[s]=0;
    ///建立一个维护最小值的优先队列
    priority_queue<PII,vector<PII>,greater<PII>>heap;
    heap.push({0,s});///起始点放入队列
    while(heap.size()){
        auto t=heap.top();///最小值
        heap.pop();
        int ver=t.second,d=t.first;
        if(st[ver]) continue;///该点更新
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=edge[i].ne){
            int j=edge[i].e;
            if(dis[j]>d+1){
                dis[j]=d+1;
                heap.push({dis[j],j});
            }
        }
    }
}

void dijkstra1(int s){
    memset(dis1,0x3f,sizeof dis1);
    memset(st,0,sizeof st);
    dis1[s]=0;
    ///建立一个维护最小值的优先队列
    priority_queue<PII,vector<PII>,greater<PII>>heap;
    heap.push({0,s});///起始点放入队列
    while(heap.size()){
        auto t=heap.top();///最小值
        heap.pop();
        int ver=t.second,d=t.first;
        if(st[ver]) continue;///该点更新
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=edge[i].ne){
            int j=edge[i].e;
            if(dis1[j]>d+1){
                dis1[j]=d+1;
                heap.push({dis1[j],j});
            }
        }
    }
}
int g[1100][1100];
int main()
{
    memset(h,-1,sizeof h);idx=0;
    n=read,m=read,s=read,t=read;
    rep(i,1,m){
        int u=read,v=read;
        add(u,v);add(v,u);
        g[u][v]=g[v][u]=1;
    }
    dijkstra(s);
    dijkstra1(t);
    int res=0;
    for(int i=1;i<n;i++)
        for(int j=i+1;j<=n;j++)
            if(!g[i][j]){
                int d1=dis[i]+dis1[j]+1,d2=dis[j]+dis1[i]+1;
                if(d1>=dis[t]&&d2>=dis[t]) res++;
            }
    printf("%d
",res);
    return 0;
}