68pts求助 玄一棺
查看原帖
68pts求助 玄一棺
777848
superkkkeee楼主2024/10/14 19:45

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;
}
2024/10/14 19:45
加载中...