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[],dis[][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=,f=; char a=getchar();
     while(a<'' || a>'') {if(a=='-') f=-; a=getchar();}
     while(a>='' && a<='') x=x*+a-'',a=getchar();
     return x*f;
 }
 struct edges{
     int fr,to,v,next;
 }e[],e1[];
 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[][u]>dis[][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]]=; memset(y,,sizeof(y));q.push((data){a[x],});
     int s,to;
     while(!q.empty()){
         s=q.top().num; q.pop();
         if(y[s]) continue; y[s]=;
         for(int i=head[s];i>=;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>=;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,,sizeof(dis));
     memset(head,-,sizeof(head));
     memset(head1,-,sizeof(head1));
     for(int i=;i<=;i++) a[i]=read();
     for(int u,v,c,i=;i<=m;i++)
     u=read(),v=read(),c=read(),inser(u,v,c);
     for(int i=;i<=;i++) dj(i);
     for(int i=;i<cnt;i+=){
         int u=e[i].fr,v=e[i].to;
         if((dis[][u]+e[i].v+dis[][v]==dis[][a[]] || dis[][v]+e[i].v+dis[][u]==dis[][a[]]) && (dis[][u]+e[i].v+dis[][v]==dis[][a[]] || dis[][v]+e[i].v+dis[][u]==dis[][a[]]))
         ins(u,v,e[i].v);
     }
     for(int i=;i<=n;i++)
     if(!r[i]) topsort(i);
     printf("%d\n",fin);
     return ;
 }