Invitation Cards(链式前向星存图+最短路)

原题链接https://vjudge.net/contest/356222#problem/J

题意为有n个点,给出单向路径,求1到每个点的最短路以及回到1的最短路。
根据题意可以知道,n的值<=1000000,此时二维数组显然不能用来存图了,在这里使用链式前向星存图。

链式前向星存图:

struct node
{
	long long v;//终点
	long long w;//权值
	long long next;//记录与当前起始地相连的上一条路径的地址
} stu[3000005];//储存题目给出的路径
long long head[3000005];//记录每一个起点的最后一个终点,

建图

void add(long long u,long long v,long long w,long long i)
{
	stu[i].v=v;
	stu[i].w=w;
	stu[i].next=head[u];
	head[u]=i;
}

链式前向前星在储存时将同一点的所有路径储存在以head为终点,结构体建立的链表中,在使用最短路时对于一个点a,以head[a]为起点,遍历链表即可。

在这里插入图片描述
遍历点a的所有路径

for(i=head[a];i!=0;i=stu[i].next)
{

}

了解链试前向星之后题目就较为简单,正反存图,spfa求1的最短路,将所有点的最短路相加即可。
完整代码:

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <queue>
#include <stack>
#include <map> 
using namespace std;
const int INF = 0x3f3f3f3f;
long long n,m;
long long i,j;	
long long sum1,sum2;
string a,b,c;
long long ans;
map<string,long long>name;
struct node
{
	long long u;
	long long v;
	long long w;	
	long long next;
} stu[3000005],sum[3000005];//sum记录路径,stu使用链式前向星
long long head[3000005];
long long dis[3000005];
queue<long long>q;
bool vis[3000005];
void add(long long u,long long v,long long w,long long i)
{
	stu[i].v=v;
	stu[i].w=w;
	stu[i].next=head[u];
	head[u]=i;
}
void spfa(long long s)
{
	for(i=1;i<=n;i++)
	{
		vis[i]=false;
		dis[i]=INF;
	}
	dis[s]=0;
	//cout<<n<<endl;
	q.push(s);
	vis[s]=true;
	long long u,v,w;
	while(!q.empty())
	{
		long long x=q.front();
		q.pop();
		vis[x]=false;
		for(i=head[x];i!=0;i=stu[i].next)
		{
			v=stu[i].v;
			//printf("%lld\n",v);
			w=stu[i].w;
			if(dis[v]>dis[x]+w)
			{
				dis[v]=dis[x]+w;	
				if(!vis[v])
				{
					q.push(v);
					vis[v]=true;
				}
			}
		}
	}
	for(i=1;i<=n;i++)//1到所有点的最短路都要加起来
	{
		ans+=dis[i];
	}
	return ;
 } 
int main()
{
	long long t;
	scanf("%lld",&t);
	while(t--)
	{
		scanf("%lld %lld",&n,&m);
		long long a,b,c;
		ans=0;
		for(i=0;i<=m;i++)
		{
			head[i]=0;
		}
		for(i=1;i<=m;i++)
		{
			scanf("%lld %lld %lld",&sum[i].u,&sum[i].v,&sum[i].w);//正向存图,同时sum为反向存图记录路径,
			add(sum[i].u,sum[i].v,sum[i].w,i);
		}
		/*for(i=head[2];i!=0;i=stu[i].next)
		{
			printf("%lld ",stu[i].w);
		}
		cout<<endl;
		for(i=1;i<=m;i++)
		{
			printf("%lld ",head[i]);
		}
		cout<<endl;*/
		spfa(1);
		for(i=0;i<=m;i++)
		{
			head[i]=0;
		}
		for(i=1;i<=m;i++)
		{
			add(sum[i].v,sum[i].u,sum[i].w,i);//使用之前记录过的sum,反向建立链式前向星,进行计算。
		}
		spfa(1);
		printf("%lld\n",ans);
	}
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值