POJ 1511 Invitation Cards

题目链接:http://poj.org/problem?id=1511

解析:就是邻接表+spfa算法,这题用vector+spfa超时(白费我码了好长时间了( ‵o′)凸),然后我就开始自学了静态链表。

有关spfa算法的话,如果不会,可以参考http://www.cnblogs.com/devtang/archive/2011/08/25/spfa.html大牛的blog

静态链表的话开三个数组,fir[i],nex[i],e[i],其中fir[i]存的是第i个点的最新输入的一条边在e[]中的下标,nex[i]存的是与第i个边相同起点的边的下标,e[i]是第i条边,里面是个结构,包括起点,终点,权值。

e1[i].x = a;e1[i].y = b;e1[i].len = c; nex1[i] = fir1[e1[i].x];fir1[e1[i].x] = i;这一段代码的话就是来维护fir1和nex1这两个数组。每次输入一个边,如果该边是从a节点出来的第一条边(即没有与之同起点的边),那么fir1[a]==-1,然后nex1[i] = fir1[a] = -1;如果已经存在了从a点出来的边,那么fir1[a]肯定就是上一条从a点出来边的下标,就把他赋值给nex1[i],代表第i条边与上一条边有关系,然后nex1[i]存的就是与第i条边同起点边的下标,而fir1[a]就需要更新为最新的这个边的下标了。

这样的话遍历的时候就会能有第t个点,到fir1[t]求出来以t为起点的一条边i,然后nex1[i]就是与第i边同起点的另外一条边j,然后再找nex1[j],一直找下去就能把t点能到的点全都遍历了。



//spfa算法+静态邻接表。
//求各点到1的距离的时候是可以把边反向,然后再次用spfa来求1到各点的距离
#include<stdio.h>
#include<string.h>
#include<queue>
#define INF 1001000
#define Max 10000000000000	//因为每个边10亿,不知道有几个边,所以开大点
using namespace std;

//下面所有数组全部都是双份的╮(╯▽╰)╭,后面有1的是正向边,2的反向边
struct edge
{
	int x,y,len;
}e1[INF],e2[INF];
long long dist[INF],dist2[INF];
int N,p,q;
int fir1[INF],fir2[INF],nex1[INF],nex2[INF],flag1[INF],flag2[INF];
queue<int>Q;

void spfa()
{
	//正向边的spfa
	while(!Q.empty())
		Q.pop();
	for(int i = 1 ; i <= p; i++)
		dist[i] = Max;
	memset(flag1,0,sizeof(flag1));
	flag1[1] = 1;//把1点放入队列
	Q.push(1);
	dist[1] = 0;//同时把dist[1]归0
	while(!Q.empty())
	{
		int t = Q.front();
		Q.pop();
		for(int i = fir1[t]; i >= 1 ;i = nex1[i])	//i存的是由点t所能到达边的下标,next中存的是与第i个边起点相同的边的下标
		{
			if(dist[e1[i].y] > dist[t] + e1[i].len )
			{
				dist[e1[i].y] = dist[t] + e1[i].len ;
				if(!flag1[e1[i].y])
				{
					Q.push(e1[i].y );
					flag1[e1[i].y] = 1;
				}
			}
		}
		flag1[t] = 0;
	}



	//反向边的spfa
	while(!Q.empty())
		Q.pop();
	for(int i = 1 ; i <= p; i++)
		dist2[i] = Max;
	memset(flag2,0,sizeof(flag2));
	flag2[1] = 1;
	Q.push(1);
	dist2[1] = 0;
	while(!Q.empty())
	{
		int t = Q.front();
		Q.pop();
		for(int i = fir2[t]; i >= 1 ;i = nex2[i])	//i存的是由点t所能到达边的下标,next中存的是与第i个边起点相同的边的下标
		{
			if(dist2[e2[i].y] > dist2[t] + e2[i].len )
			{
				dist2[e2[i].y] = dist2[t] + e2[i].len ;
				if(!flag2[e2[i].y])
				{
					Q.push(e2[i].y );
					flag2[e2[i].y] = 1;
				}
			}
		}
		flag2[t] = 0;
	}
}

int main()
{
	scanf("%d",&N);
	while(N--)
	{
		memset(dist,0,sizeof(dist));
		memset(dist2,0,sizeof(dist2));
		memset(fir1,-1,sizeof(fir1));
		memset(fir2,-1,sizeof(fir2));
		memset(nex1,0,sizeof(nex1));
		memset(nex2,0,sizeof(nex2));
		memset(flag1,0,sizeof(flag1));
		memset(flag2,0,sizeof(flag2));
		scanf("%d%d",&p,&q);
		for(int i = 1; i <= q; i ++)
		{
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			e1[i].x = a;e1[i].y = b;e1[i].len = c;	//存正向的边
			nex1[i] = fir1[e1[i].x];
			fir1[e1[i].x] = i;


			e2[i].x = b;e2[i].y = a;e2[i].len = c;	//存反向的边
			nex2[i] = fir2[e2[i].x];
			fir2[e2[i].x] = i;
		}
		spfa();

		long long sum = 0;
		for(int i = 1 ; i <= p; i ++)
		{
			sum+=dist[i];
			sum+=dist2[i];
		}
		printf("%lld\n",sum);
	}
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值