poj1511 spfa

题意：有N个节点，节点间的道路为单向。求从第一个节点到所有其他节点以及从所有其他节点到第一个节点的最短路之和。

思路：spfa算法。需要建图两次。

#include <stdio.h>
#include <string.h>
#define N 1000005
#define INF 0x3fffffff
int map[N][3];
struct edge{
	int y,w,next;
}e[N];
int first[N],dis[N],visited[N],res[N],q[N];
int n,m,top,T;
void init(){
	int i;
	top = 0;
	memset(visited,0,sizeof(visited));
	memset(first,-1,sizeof(first));
	for(i = 1;i<=n;i++)
		dis[i] = INF;
}
void add(int x,int y,int w){
	e[top].y = y;
	e[top].w = w;
	e[top].next = first[x];
	first[x] = top++;
}
int relax(int a,int b,int w){
	if(dis[a]+w < dis[b]){
		dis[b] = dis[a]+w;
		return 1;
	}
	return 0;
}
void spfa(){
	int i,front,rear;
	front = rear = -1;
	q[++rear] = 1;
	visited[1] = 1;
	dis[1] = 0;
	while(front < rear){
		int pos = q[++front];
		for(i = first[pos];i!=-1;i=e[i].next)
			if(1==relax(pos,e[i].y,e[i].w) && !visited[e[i].y]){
				q[++rear] = e[i].y;
				visited[e[i].y] = 1;
			}
	}
	for(i = 2;i<=n;i++)
		res[i] += dis[i];
}
int main(){
	freopen("a.txt","r",stdin);
	scanf("%d",&T);
	while(T--){
		int i;
		__int64 sum = 0;
		memset(res,0,sizeof(res));
		scanf("%d %d",&n,&m);
		for(i = 0;i<m;i++)//需要两次建图，所以先将图存起来
			scanf("%d %d %d",&map[i][0],&map[i][1],&map[i][2]);
		init();
		for(i = 0;i<m;i++)
			add(map[i][0],map[i][1],map[i][2]);
		spfa();
		init();
		for(i = 0;i<m;i++)
			add(map[i][1],map[i][0],map[i][2]);
		spfa();
		for(i = 2;i<=n;i++)
			sum+=res[i];
		printf("%I64d\n",sum);
	}
	return 0;
}