【最短路+容斥】arc090E Avoiding Collision

题目：https://vjudge.net/contest/464115#problem/G

题意：

一个$n$个点$m$条边的无向图，每条边有一个边权$w$。有一个起点$s$和终点$t$，求从$s$出发到$t$和从$t$出发到$s$的不相遇的两条最短路径的路径对数。

思路：

$dijkstra$ 求出起点到$i$的距离$dis[0][i]$，终点到$i$的距离$dis[1][i]$，起点到$i$最短路的方案数$cnt[0][i]$，终点到$i$的最短路的方案数$cnt[1][i]$。

求出所有最短路径数，任选两条$C_{cnt[0][t]}^2\ast 2$的方案数减去相遇的方案数就是答案

考虑在点上相遇

当$dis[0][i]==dis[1][i]$&&$dis[0][i]==2\ast dis[0][t]$时，在点$i$相遇的方案为$C_{cnt[0][i]\ast cnt[1][i]}^2\ast 2$

考虑在边上相遇

设边$i$的端点$u_i$，$v_i$，$u_i$理起点更近，边权为$w_i$。

当$dis[0][u]+dis[1][v]+w==dis[0][t]$&&$dis[0][u]\ast 2<dis[0][t]$&&$dis[1][v]\ast 2<dis[0][t]$时，在边$i$相遇的方案数为$C_{cnt[0][u]\ast cnt[1][v]}^2\ast 2$

代码：

#include<bits/stdc++.h>
#define MAXN 100005
#define INF 0x3f3f3f3f
#define pb push_back

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,int>P;
vector<P>G[MAXN];
const int mod=1e9+7;

struct Edge
{
    int u,v,w;
}edge[MAXN*2];

ll dis[2][MAXN],cnt[2][MAXN];
int vis[MAXN];
void dijkstra(int id,int s)
{
    memset(dis[id],60,sizeof(dis[id]));
    memset(vis,0,sizeof(vis));
    dis[id][s]=0;
    cnt[id][s]=1;
    priority_queue<P,vector<P>,greater<P> >q;
    q.push(P(0,s));
    while(!q.empty())
    {
        P p=q.top();q.pop();
        int x=p.second;
        if(vis[x])continue;
        vis[x]=1;
        for(P e:G[x])
        {
            int to=e.second;//cout<<x<<' '<<to<<' '<<dis[id][to]<<endl;
            ll w=e.first;
            if(dis[id][to]>dis[id][x]+w)
            {
                dis[id][to]=dis[id][x]+w;
                cnt[id][to]=cnt[id][x];
                q.push(P(dis[id][to],to));
            }
            else if(dis[id][to]==dis[id][x]+w)
                cnt[id][to]=(cnt[id][to]+cnt[id][x])%mod;
        }
    }
}

int main(){
    int n,m;scanf("%d%d",&n,&m);
    int s,t;scanf("%d%d",&s,&t);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;scanf("%d%d%d",&u,&v,&w);
        edge[i]={u,v,w};
        G[u].pb(P(w,v));
        G[v].pb(P(w,u));
    }
    dijkstra(0,s);
    dijkstra(1,t);
    ll ans=cnt[0][t]*(cnt[0][t]-1)%mod;//cout<<ans<<endl;
    for(int i=1;i<=n;i++)
    {
        if(dis[0][i]==dis[1][i]&&dis[0][i]*2==dis[0][t])
        {
            ll tmp=cnt[0][i]*cnt[1][i]%mod;
            ans=(ans-tmp*(tmp-1)%mod+mod)%mod;
        }
    }
    for(int i=1;i<=m;i++)
    {
        int u=edge[i].u,v=edge[i].v,w=edge[i].w;
        if(dis[0][u]>dis[0][v])swap(u,v);
        if(dis[0][u]+dis[1][v]+w==dis[0][t]&&dis[0][u]*2<dis[0][t]&&dis[1][v]*2<dis[0][t])
        {
            ll tmp=cnt[0][u]*cnt[1][v]%mod;
            ans=(ans-tmp*(tmp-1)%mod+mod)%mod;
        }
    }
    printf("%lld\n",ans);
    return 0;
}