题意:求s1到t1和s2到t2的最短路中重合的最长部分(最短路不止一条)
题解
- 最短路大家都会吧?
- 这道题需要把最短路从原图中挖出来。判断一条边在不在最短路上的方法是:如果AB有一条边,且A到s的距离+边AB的长度+B到t的距离==s到t的距离,那么边AB就在s到t的最短路上
- 注意建的新图是有向图
- 两个人反向走的边也可以算作重合>_>
代码
这个代码不知道哪里TLE了 QAQ
大神求教
// by spli
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstdio>
using namespace std;
const int N=1510;
const int inf=0x3f3f3f3f;
int n,m;
int s1,t1,s2,t2;
struct node{
int f,to,nxt,val;
}e[N*N<<1],et[N*N<<1];int head[N],cnt;int ht[N],tcnt;
int diss[3][N];
int dist[3][N];
bool vis[N],init[N];
queue<int>q;
int deg[N],d[N];
int ans;
void add(int f,int t,int v){
cnt++;
e[cnt]=(node){f,t,head[f],v};
head[f]=cnt;
}
void addt(int f,int t,int v){
tcnt++;
et[tcnt]=(node){f,t,ht[f],v};
ht[f]=tcnt;
deg[t]++;
init[f]=init[t]=1;
}
void spfas(int k,int s){
for(int i=1;i<=n;++i) diss[k][i]=inf;
q.push(s);vis[s]=1;diss[k][s]=0;
while(!q.empty()){
int u=q.front();
q.pop();
vis[u]=0;
for(int i=head[u];i!=-1;i=e[i].nxt){
int v=e[i].to;
if(diss[k][v]>diss[k][u]+e[i].val){
diss[k][v]=diss[k][u]+e[i].val;
if(!vis[v]){
vis[v]=1;
q.push(v);
}
}
}
}
}
void spfat(int k,int s){
for(int i=1;i<=n;++i) dist[k][i]=inf;
q.push(s);vis[s]=1;dist[k][s]=0;
while(!q.empty()){
int u=q.front();
q.pop();
vis[u]=0;
for(int i=head[u];i!=-1;i=e[i].nxt){
int v=e[i].to;
if(dist[k][v]>dist[k][u]+e[i].val){
dist[k][v]=dist[k][u]+e[i].val;
if(!vis[v]){
vis[v]=1;
q.push(v);
}
}
}
}
}
void topsort(){
for(int i=1;i<=n;++i)
if(!deg[i]) q.push(i);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=ht[u];i;i=et[i].nxt){
int v=et[i].to;
d[v]=max(d[v],d[u]+et[i].val);
ans=max(ans,d[v]);
deg[v]--;
if(deg[v]==0) q.push(v);
}
}
}
int main(){
memset(head,-1,sizeof(head));
scanf("%d%d",&n,&m);
scanf("%d%d%d%d",&s1,&t1,&s2,&t2);
int x,y,z;
for(int i=1;i<=m;++i){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);add(y,x,z);
}
spfas(1,s1);
spfas(2,s2);
spfat(1,t1);
spfat(2,t2);
int u,v,s;
for(int i=1;i<=cnt;i+=2){
u=e[i].f,v=e[i].to,s=e[i].val;
int s1=min(diss[1][u],diss[1][v])+s+min(dist[1][u],dist[1][v]);
int s2=min(diss[2][u],diss[2][v])+s+min(dist[2][u],dist[2][v]);
if(s1==diss[1][t1]&&s2==diss[2][t2]){
if(diss[1][u]<diss[1][v]) addt(u,v,s);
else addt(v,u,s);
}
}
topsort();
cout<<ans<<endl;
return 0;
}