WA on #3 #7 #9 #11
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m;
int s,t,ss,tt;
struct node
{
int v,w;
};
vector<node> p[15001];
vector<node> e[15001];
int dis[11][15001];
int diss[15001];
bool vis[15001];
int ind[15001];
int ans;
void hs(int s,int id)
{
priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > > q;
while(!q.empty()) q.pop();
memset(vis,0,sizeof(vis));
dis[id][s]=0;
q.push({0,s});
while(!q.empty())
{
int x=q.top().second; q.pop();
if(vis[x]) continue;
vis[x]=1;
for(node xx:p[x])
{
int v=xx.v;
int w=xx.w;
if(dis[id][v]>dis[id][x]+w)
{
dis[id][v]=dis[id][x]+w;
q.push({dis[id][x],v});
}
}
}
}
void topu()
{
//for(int i=1;i<=n;i++) diss[i]=0;
queue<int> q;
while(!q.empty()) q.pop();
for(int i=1;i<=n;i++)
{
if(!ind[i]) q.push(i);
}
while(!q.empty())
{
int x=q.front(); q.pop();
for(node xx:e[x])
{
int v=xx.v;
int w=xx.w;
diss[v]=max(diss[v],diss[x]+w);
ind[v]--;
if(!ind[v]) q.push(v);
}
}
}
signed main()
{
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
cin>>n>>m>>s>>t>>ss>>tt;
memset(dis,0x7f7f7f7f,sizeof(dis));
for(int i=1;i<=m;i++)
{
int u,v,w; cin>>u>>v>>w;
p[u].push_back({v,w});
p[v].push_back({u,w});
}
hs(s,1); hs(t,2);
hs(ss,3); hs(tt,4);
for(int x=1;x<=n;x++)
{
for(node xx:p[x])
{
int v=xx.v;
int w=xx.w;
if(/*dis[1][x]+w+dis[2][v]==dis[1][t]&&dis[3][x]+w+dis[4][v]==dis[3][tt]*/dis[1][x]+w+dis[2][v]==dis[1][t]&&dis[3][x]+w+dis[4][v]==dis[3][tt])
{
e[x].push_back({v,w});
ind[v]++;
}
}
}
topu();
for(int i=1;i<=n;i++)
{
ans=max(ans,diss[i]);
}
for(int i=0;i<=n;i++) e[i].clear();
memset(diss,0,sizeof(diss)); memset(ind,0,sizeof(ind));
for(int x=1;x<=n;x++)
{
for(node xx:p[x])
{
int v=xx.v;
int w=xx.w;
/*if(dis[1][x]+w+dis[2][v]==dis[1][t]&&dis[4][x]+w+dis[3][v]==dis[3][tt])
{
e[x].push_back({v,w});
ind[v]++;
}*/
if(dis[1][x]+w+dis[2][v]==dis[1][t]&&dis[4][x]+w+dis[3][v]==dis[3][tt])
{
e[x].push_back({v,w});
ind[v]++;
}
}
}
topu();
for(int i=1;i<=n;i++)
{
ans=max(ans,diss[i]);
}
cout<<ans;
return 0;
}