bzoj1880: [Sdoi2009]Elaxia的路线

传送门
大力做两遍最短路
然后判断一条便是否在两条最短路上
之后大力拓扑排序九可以了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<queue>
#define N 1505
using namespace std;
queue<int> q;
int S,T,s,t,n,m,mp[N][N],dis[4][N],v[N];
struct edge{int to,next,v;}e[N*N*2];
int head[N],d[N],f[N],ans,tot;
void add(int x,int y,int v){
    e[++tot]=(edge){y,head[x],v};
    head[x]=tot;
}
void spfa(int s,int d){
    memset(dis[d],0x3f,sizeof(dis[d]));
    dis[d][s]=0; v[s]=1;
    q.push(s);
    while (!q.empty()){
        int x=q.front(); q.pop(); v[x]=0;
        for (int i=1;i<=n;i++)
            if (dis[d][i]>dis[d][x]+mp[x][i]){
                dis[d][i]=dis[d][x]+mp[x][i];
                if (!v[i]) v[i]=1,q.push(i);
            }
    }
}
void topo(){
    for (int i=1;i<=n;i++) if (!d[i]) q.push(i);
    while (!q.empty()){
        int x=q.front(); q.pop();
        ans=max(ans,f[x]);
        for (int i=head[x];i;i=e[i].next){
            f[e[i].to]=max(f[e[i].to],f[x]+e[i].v);
            d[e[i].to]--;
            if (!d[e[i].to]) q.push(e[i].to);
        }
    }
}
int main(){
    scanf("%d%d%d%d%d%d",&n,&m,&S,&T,&s,&t);
    memset(mp,0x3f,sizeof(mp));
    for (int i=1;i<=n;i++) mp[i][i]=0;
    for (int i=1,a,b,c;i<=m;i++){
        scanf("%d%d%d",&a,&b,&c);
        mp[a][b]=mp[b][a]=c;
    }
    spfa(S,0); spfa(T,1);
    spfa(s,2); spfa(t,3);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++) if (i!=j)
            if (dis[0][i]+mp[i][j]+dis[1][j]==dis[0][T])
                if (dis[2][i]+mp[i][j]+dis[3][j]==dis[2][t])
                    add(i,j,mp[i][j]),d[j]++;
    topo();
    memset(f,0,sizeof(f));
    memset(head,0,sizeof(head));
    tot=0;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++) if (i!=j)
            if (dis[0][i]+mp[i][j]+dis[1][j]==dis[0][T])
                if (dis[3][i]+mp[i][j]+dis[2][j]==dis[2][t])
                    add(i,j,mp[i][j]),d[j]++;
    topo();
    printf("%d\n",ans); 
}