BZOJ1880: [Sdoi2009]Elaxia的路线

题意：求最短路最长公共距离。

考虑每一条边，如果满足dis(s1,u)+len+dis(v,t1)==dis(s1,t1) && dis(s2,u)+len+dis(v,t2)==dis(s2,t2) 则该边在公共最短路上

拓扑排序dp下即可

#include<bits/stdc++.h>
using namespace std;
#define N 1505
#define INF 1e9
int n,m,a[5],dis[5][N],cnt,cnt1,head[N],head1[N],r[N],ans[N],fin;
bool y[N];
struct data{
    int num,v;
    bool operator < (const data& w)const{
       return v>w.v;
    }
};
priority_queue<data>q;
inline int read(){
    int x=0,f=1; char a=getchar();
    while(a<'0' || a>'9') {if(a=='-') f=-1; a=getchar();}
    while(a>='0' && a<='9') x=x*10+a-'0',a=getchar();
    return x*f;
} 
struct edges{
    int fr,to,v,next;
}e[500005],e1[500005];
void inser(int u,int v,int c){
    e[cnt]=(edges){u,v,c,head[u]};head[u]=cnt++;
    e[cnt]=(edges){v,u,c,head[v]};head[v]=cnt++;
}
void ins(int u,int v,int c){
    if(dis[1][u]>dis[1][v]) swap(u,v);
    e1[cnt1]=(edges){u,v,c,head1[u]};head1[u]=cnt1++; r[v]++;
}
void dj(int x){
    dis[x][a[x]]=0; memset(y,0,sizeof(y));q.push((data){a[x],0});
    int s,to;
    while(!q.empty()){
        s=q.top().num; q.pop(); 
        if(y[s]) continue; y[s]=1;
        for(int i=head[s];i>=0;i=e[i].next){
        if(y[e[i].to]) continue; to=e[i].to;
        if(dis[x][to]>dis[x][s]+e[i].v) dis[x][to]=dis[x][s]+e[i].v,q.push((data){to,dis[x][to]});
        }
    }
}
void topsort(int x){
    r[x]--;
    for(int i=head1[x];i>=0;i=e1[i].next){
        ans[e1[i].to]=max(ans[e1[i].to],ans[x]+e1[i].v);
        fin=max(ans[e1[i].to],fin);
        r[e1[i].to]--; if(!r[e1[i].to]) topsort(e1[i].to);
    }
}
int main(){
    n=read(); m=read(); 
    memset(dis,34,sizeof(dis)); 
    memset(head,-1,sizeof(head));
    memset(head1,-1,sizeof(head1));
    for(int i=1;i<=4;i++) a[i]=read();
    for(int u,v,c,i=1;i<=m;i++)
    u=read(),v=read(),c=read(),inser(u,v,c);
    for(int i=1;i<=4;i++) dj(i);
    for(int i=0;i<cnt;i+=2){
        int u=e[i].fr,v=e[i].to;
        if((dis[1][u]+e[i].v+dis[2][v]==dis[1][a[2]] || dis[1][v]+e[i].v+dis[2][u]==dis[1][a[2]]) && (dis[3][u]+e[i].v+dis[4][v]==dis[3][a[4]] || dis[3][v]+e[i].v+dis[4][u]==dis[3][a[4]]))
        ins(u,v,e[i].v);
    }
    for(int i=1;i<=n;i++)
    if(!r[i]) topsort(i);
    printf("%d
",fin);
    return 0;
}