10分求助!!!
查看原帖
10分求助!!!
133034
BetrayalObedience楼主2022/8/8 14:06
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<queue>
using namespace std;
int n,m,head[1000001],cnt,dis[1000001],ans,dis1[1000001],cnt1,head1[1000001];
bool vis[1000001],vis1[1000001];
struct node
{
	int to,nxt,val;
}a[1000001];
struct node1
{
	int to,nxt,val;
}b[1000001];
void add(int u,int v,int w)
{
	a[++cnt].nxt=head[u];
	head[u]=cnt;
	a[cnt].to=v;
	a[cnt].val=w;
}
void add1(int u,int v,int w)
{
	b[++cnt1].nxt=head1[u];
	head1[u]=cnt1;
	b[cnt1].to=v;
	b[cnt1].val=w;
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
		add1(v,u,w);
	}
	for(int i=1;i<=n;i++)
	{
		dis[i]=1e9;
		vis[i]=0;
	}
	queue<int> q;
	q.push(1);
	vis[1]=1;
	dis[1]=0;
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		vis[u]=1;
		for(int i=head[u];i;i=a[i].nxt)
		{
			int v=a[i].to;
			if(dis[v]>dis[u]+a[i].val)
			{
				dis[v]=dis[u]+a[i].val;
				if(!vis[v])
				{
					vis[v]=1;
					q.push(v);
				}
			}
		}
	}
	for(int i=1;i<=n;i++) ans+=dis[i];
	for(int i=1;i<=n;i++)
	{
		dis1[i]=1e9;
		vis1[i]=0;
	}
	queue<int> qq;
	qq.push(1);
	dis1[1]=0;
	vis1[1]=1;
	while(!qq.empty())
	{
		int u=qq.front();
		qq.pop();
		vis1[u]=1;
		for(int i=head1[u];i;i=b[i].nxt)
		{
			int v=b[i].to;
			if(dis1[v]>dis1[u]+b[i].val)
			{
				dis1[v]=dis1[u]+b[i].val;
				if(!vis1[v])
				{
					vis1[v]=1;
					qq.push(v);
				}
			}
		}
	}
	int ans1=0;
	for(int i=1;i<=n;i++) ans1+=dis1[i];
	cout<<ans+ans1;
}
2022/8/8 14:06
加载中...