POJ1511 Invitation Cards（SPFA+逆图）

最新推荐文章于 2019-09-07 11:58:00 发布

xtttgo

最新推荐文章于 2019-09-07 11:58:00 发布

阅读量271

点赞数

CC 4.0 BY-SA版权

分类专栏：最短路

本文链接：https://blog.youkuaiyun.com/xtttgo/article/details/48102799

最短路专栏收录该内容

8 篇文章

订阅专栏

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

题意：

求从1出发到每个点，再从每个点出发到1的最短路

从1到每个点的最短路，用一次spfa就能求出来了，但是从每个点到1的就很难求

所以存了两幅图，一副是正常方向的图，一副是相反的逆图，再从1出发，求一次逆图的最短路

就可以求出从每个点到1的距离了

而且，因为这题数据比较大，所以我原本准备用vector来存边，结果会超时，所以从其他博客中学习，改用链表的形式来存边，能节省很多空间和时间

非常好的方法

代码：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<string>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<ctime>
using namespace std;
const int INF = 1e9 + 9;
const int MAXN = 1000000 + 10;
int n, m;
struct node{
	int to, l;
	int next;
}edge[2][MAXN];//0是正向图，1是逆向图
bool vis[MAXN];
int d[MAXN];
int tol[2];
int head[2][MAXN];
void addedge(int s, int e, int l, int k)//用k标记是正图还是逆图
{
	int &tot = tol[k];
	edge[k][tot].to = e;
	edge[k][tot].l = l;
	edge[k][tot].next = head[k][s];
	head[k][s] = tot++;
}
void Spfa(int st, int k)
{
	int i;
	for (i = 1; i <= n; i++)
	{
		d[i] = INF;
		vis[i] = false;
	}
	d[st] = 0;
	vis[st] = true;
	queue<int> Q;
	Q.push(st);
	while (!Q.empty())
	{
		int start = Q.front();
		Q.pop();
		vis[start] = false;
		for (i = head[k][start]; i != -1; i = edge[k][i].next)
		{
			if (d[start] + edge[k][i].l < d[edge[k][i].to])
			{
				d[edge[k][i].to] = d[start] + edge[k][i].l;
				if (!vis[edge[k][i].to])
				{
					vis[edge[k][i].to] = true;
					Q.push(edge[k][i].to);
				}
			}
		}
	}
}
int main()
{
//  freopen("D://input.txt", "r", stdin);
//  freopen("D://output.txt", "w", stdout);
	int T;
	scanf("%d", &T);
	while (T--)
	{
		memset(head, -1, sizeof(head));
		tol[0] = tol[1] = 0;//边数初始化为0
		int i;
		scanf("%d%d", &n, &m);
		for (i = 0; i < m; i++)
		{
			int x, y, z;
			scanf("%d%d%d", &x, &y, &z);
			addedge(x, y, z, 0);
			addedge(y, x, z, 1);
		}
		__int64 sum = 0;
		Spfa(1, 0);
		for (i = 1; i <= n; i++)
		{
			sum += d[i];
		}
		Spfa(1, 1);//两次spfa
		for (i = 1; i <= n; i++)
		{
			sum += d[i];
		}
		printf("%I64d\n", sum);
	}
//  printf("\n%.3lf\n",clock()/CLOCKS_PER_SEC);
	return 0;
}