BZOJ1880[Sdoi2009]Elaxia的路线 spfa+拓扑序

首先我们求出来每一个点到$x1,x2,y1,y1$的最短路
令$s1,s2$分别为两个人的最短路
枚举每条边如果$dis[x1][u]+dis[y1][v]+e[i].c=s1,dis[x2][u]+dis[y2][v]+e[i].c=s2$ 则在新的图中加入这条边
我们可以发现最后选出的边一定是一些链 所以最后跑拓扑序

#include<bits/stdc++.h>
#define bug(x) cout<<(#x)<<" "<<(x)<<endl
#define ll long long
using namespace std;
const int N=1505;
inline int read(){
    int 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;
}
int n,m,ans,tot=1,cnt=1,dis[4][N],sum[N],head[N],in[N]; 
bool vis[N];
queue<int>q;
struct Egde{
    int u,v,nxt,c;
}e[N*N];
void add(int u,int v,int c){
    e[++tot].v=v,e[tot].u=u,e[tot].nxt=head[u],e[tot].c=c,head[u]=tot;
    e[++tot].v=u,e[tot].u=v,e[tot].nxt=head[v],e[tot].c=c,head[v]=tot;
}
void Add(int u,int v,int c){
    e[++cnt].v=v,e[cnt].nxt=head[u],e[cnt].c=c,head[u]=cnt,in[v]++;
}
void spfa(int s,int op){
    while(!q.empty()) q.pop();memset(vis,0,sizeof vis);
    q.push(s),dis[op][s]=0,vis[s]=1;
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=head[x];i;i=e[i].nxt){
            int j=e[i].v;
            if(dis[op][j]>dis[op][x]+e[i].c){
                dis[op][j]=dis[op][x]+e[i].c;
                if(!vis[j]) vis[j]=1,q.push(j);
            }
        }
        vis[x]=0;
    }
}
void topsort(){
    while(!q.empty()) q.pop();
    for(int i=1;i<=n;i++) if(!in[i]) q.push(i);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=head[x];i;i=e[i].nxt){
            int j=e[i].v;
            sum[j]=max(sum[j],sum[x]+e[i].c);
            if(!--in[j]) q.push(j);
        }
    }
    for(int i=1;i<=n;i++) ans=max(ans,sum[i]);
    cout<<ans<<endl; 
}
int main(){
#ifdef Devil_Gary
    freopen("in.txt","r",stdin);
#endif
    memset(dis,127,sizeof dis);
    n=read(),m=read();
    int zjq1=read(),zjq2=read(),amaz1=read(),amaz2=read();
    for(int i=1,x,y,z;i<=m;i++){
        x=read(),y=read(),z=read();
        add(x,y,z);
    }
    spfa(zjq1,1),spfa(zjq2,2),spfa(amaz1,3),spfa(amaz2,0);
    int s1=dis[1][zjq2],s2=dis[3][amaz2];
    memset(head,0,sizeof head);
    for(int i=2;i<=tot;i+=2){
        int u=e[i].u,v=e[i].v;
        if(min(dis[1][u],dis[1][v])+min(dis[2][u],dis[2][v])+e[i].c==s1&&min(dis[3][u],dis[3][v])+min(dis[0][u],dis[0][v])+e[i].c==s2)
            if(dis[1][u]<dis[1][v]) Add(u,v,e[i].c);
            else Add(v,u,e[i].c);
    }
    topsort();
    return 0;
}