poj1511 spfa

最新推荐文章于 2020-07-26 22:33:05 发布

dumeichen

最新推荐文章于 2020-07-26 22:33:05 发布

阅读量362

点赞数

CC 4.0 BY-SA版权

分类专栏：图论——最短路

本文链接：https://blog.youkuaiyun.com/dumeichen/article/details/37601333

图论——最短路专栏收录该内容

27 篇文章

订阅专栏

题意：有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;
}