您的位置:首页 > 其它

POJ 1511 Invitation Cards

2013-07-06 13:45 190 查看
题目链接: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);
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  spfa 静态邻接表