HDU 4725 —— The Shortest Path in Nya Graph

原题：http://acm.hdu.edu.cn/showproblem.php?pid=4725

题意：有n个点，n个层，相邻层之间花费为c；有额外的m条路，这m条路是直接点与点相连的，花费为w；问从1走到n的最小花费，若无法到达，则输出-1；

注意：一层可以有多个点，同一层中，点与点的距离不是0，而是w；

思路：我们将点拆成3n，1~n表示的就是点，另外2n则是用来表示层，每个层拆成两个点 n+2*i 和 n+2*i-1；为什么将层拆成两个点而不是一个点呢？因为同一层中的两给点之间的距离是w而不是0，假设层为A，该层上有点1和2，那么A和1，A和2之间的距离均为0，如果我们只将层表示为一个点，那么该点到1有双向路径，边权均为0；同样，该点到2有双向路径，边权均为0，这样的话，从1走到2就可以通过该点，而花费为0，但实际上，1到2的花费应该是w。所以我们将层拆成两个点，一个点出，一个点入，建单向图；

建图：点在层上—— i -> n+2*x-1，n+2*x -> i     边权为0； 层与层之间 —— n+2*i-1 -> n+2*(i+1)，n+2*(i+1)-1 -> n+2*i     边权为c；

点与点之间 —— u -> v，v -> u      边权为w；

用优先队列优化dijkstra；

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<string>
#define inf 1e9
using namespace std;
const int maxn = 1e5+5;
int cas, T = 0;
bool vis[maxn*3];
int dis[maxn*3];

struct Edge
{
	int v;
	int cost;	
	Edge(int _v = 0, int _cost = 0):v(_v), cost(_cost){}
};

struct node
{
	int v, c;
	node(int _v = 0, int _c = 0):v(_v), c(_c){}
	bool operator < (const node &a) const
	{
		return c > a.c;
	}
};

vector<Edge>vec[maxn*3];
void Dijkstra(int n, int st)
{
	for(int i = 1;i<=n;i++)	dis[i] = inf;
	memset(vis, false, sizeof vis);
	priority_queue<node>q;
	while(!q.empty())	q.pop();
	dis[st] = 0;
	q.push(node(st, 0));
	while(!q.empty())
	{
		node now = q.top();
		q.pop();
		int u = now.v;
		if(vis[u])	continue;
		vis[u] = true;
		for(int i = 0;i<vec[u].size();i++)
		{
			int v = vec[u][i].v;
			int c = vec[u][i].cost;
			if(!vis[v] && dis[v] > dis[u] + c)
			{
				dis[v] = dis[u] + c;
				q.push(node(v, dis[v]));
			}
		}
	}
}

void add(int u, int v, int w)
{
	vec[u].push_back(Edge(v, w));
}

int main()
{
	scanf("%d", &cas);
	while(cas--)
	{		
		int n, m, c;
		scanf("%d%d%d", &n, &m, &c);
		for(int i = 1;i<=3*n;i++)	vec[i].clear();
		for(int i = 1;i<=n;i++)
		{
			int x;
			scanf("%d", &x);
			add(i, n+2*x-1, 0);
			add(n+2*x, i, 0);
		}
		for(int i = 1;i<n;i++)
		{
			add(n+2*i-1, n+2*(i+1), c);
			add(n+2*(i+1)-1, n+2*i, c);
		}
		while(m--)
		{
			int u, v, w;
			scanf("%d%d%d", &u, &v, &w);
			add(u, v, w);
			add(v, u, w);
		}
		Dijkstra(3*n, 1);
		if(dis[n] == inf)	printf("Case #%d: -1\n", ++T);
		else		printf("Case #%d: %d\n", ++T, dis[n]);
	}
	return 0;
}