(POJ - 1511)Invitation Cards(最短路+反向建边)

题目链接：1511 -- Invitation Cards

简化版题意：T组样例，每组样例先给出点的个数n和边的个数m,然后给出m条边的信息a,b,c表示a到b有一条长为c的单向边，现在我们每次都需要从1号点跑到其他点然后再跑回来，问我们最少需要走的距离。

这道题需要注意的一点是题目中所给的边是单向边，如果是双向边的话那么1到其他点所需的最短距离与从其他点回来的最短距离是相等的，就是我们只需要从1号点跑一个最短路，然后加上所有距离的二倍即可。

我们求的是1号点到其他所有点的最短距离的和+所有其他点到1号点的最短距离和，我们把答案分成这两部分求，先求1号点到其他所有点的最短距离的和，这个比较简单，就是从1号点跑一个最短路求出1号点到其他点的最短路及，然后就把他们加起来即可，关键是求所有其他点到1号点的最短距离和，我们不可能从每个点都跑一遍最短路，这样肯定是会超时的，这里就需要用一个技巧了，就是我们把所有的边反向建立，然后再从1号点跑一个最短路，这样再把1号点到所有点的最短距离加起来就得到了所有其他点到1号点的最短距离和，原理是什么呢？我们像，假如原来2到1有一条长度为5的边，而我们建反向边建立的是从1到2有一条长度为5的边，如果1到2的最短距离包含此条边，这时候最短路径中从1到2所走的每一步都是反向的，也就是说可以正好从2到1按照反向边走回来，所以这样就得到了最短距离。这个技巧比较巧妙，建议好好理解一下。

下面是代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
const int N=1000003;
typedef pair<int,int>PII;
int h[N],e[N*2],ne[N*2],w[N*2],idx,d[N];
int u[N],v[N],t[N];
bool vis[N];
void add(int x,int y,int z)
{
	e[idx]=y;
	w[idx]=z;
	ne[idx]=h[x];
	h[x]=idx++;
}
void init()
{
	memset(h,-1,sizeof h);
	idx=0;
}
void dijkstra(int x)
{
	priority_queue<PII,vector<PII>,greater<PII> >q;
	memset(d,0x3f,sizeof d);
	memset(vis,false,sizeof vis);
	d[x]=0;
	q.push({d[x],x});
	while(!q.empty())
	{
		int begin=q.top().second;
		q.pop();
		if(vis[begin]) continue;
		vis[begin]=true;
		for(int i=h[begin];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(d[j]>d[begin]+w[i])
			{
				d[j]=d[begin]+w[i];
				q.push({d[j],j});
			}
		} 
	}
}
int main()
{
	int T,n,m;
	cin>>T;
	while(T--)
	{
		init();
		scanf("%d%d",&n,&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&u[i],&v[i],&t[i]);
			add(u[i],v[i],t[i]);
		}
		dijkstra(1);
		long long ans=0;
		for(int i=1;i<=n;i++)
			ans+=d[i];
		init();
		for(int i=1;i<=m;i++)
			add(v[i],u[i],t[i]);
		dijkstra(1);
		for(int i=1;i<=n;i++)
			ans+=d[i];
		printf("%lld\n",ans);
	}
	return 0;
}