POJ 3662 Telephone Lines

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

英文题面，大概意思就是， 有一个无向图，n个点p条边，其中你可以选择最多k条免费边，从起点1到终点n的最远距离是路径上的边中，除去选择的免费边以外的所有边中最长的一条边，求此限制下的最短路。

思路：

可以用分层图解决。首先，同一层次的点，例如x，p和y，p（表示从1到x，用了p条免费边，从1到y也用了p条，也就是说当前这个边x-y不是免费边），他们之间应该有一条边，长度为z，如果是x，p和y，p+1，就应该有一个长为0的边，相当于当前这个边是免费边。建图，然后跑SPFA，最后结果就是（n，p）（从1到n用了p条免费边），这个思想其实也算得上是动态规划的思想了。%%%lydrainbow

有一个点很奇怪，内存开了15mb，内存限制64mb，本来应该不会MLE，博主习惯上将表头置为-1，因此会用memset初始化表头，但是加上初始化就MLE了，莫名其妙。 

#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
#define mem(a, b) memset(a, b, sizeof a)
const int N = 1006, P = 20006, INF = 0x3f3f3f3f;
int n, m, k, cnt;
int head[N * N], nex[N * P], to[N * P], ed[N * P];
void add(int a, int b, int c){
	to[++cnt] = b;
	ed[cnt] = c;
	nex[cnt] = head[a];
	head[a] = cnt;
}
int d[N * N];
bool vis[N * N];
queue<int > q;
int spfa(){
	mem(d, 0x3f);
	mem(vis, 0);
	d[1] = 0;
	vis[1] = 1;
	q.push(1);
	while (!q.empty()){
		int t = q.front();
		q.pop();
		vis[t] = 0;
		for (int i = head[t]; i; i = nex[i]){
			int y = to[i];
			int c = ed[i];
			if (d[y] > max(d[t], c)){
				d[y] = max(d[t], c);
				if (vis[y] == 0){
					q.push(y);
					vis[y] = 1;
				}
			}
		}
	}
	return d[n * k + n];
}
int main()
{
	ios::sync_with_stdio(0);
	cnt = 0;
	mem(head, -1);
	cin >> n >> m >> k;
	for (int i = 0; i < m; i++){
		int x, y, z;
		cin >> x >> y >> z;
		for (int i = 0; i <= k; i++){
			add(i * n + x, i * n + y, z);
			add(i * n + y, i * n + x, z);
		}
		for (int i = 0; i < k; i++){
			add(i * n + x, (i + 1) * n + y, 0);
			add(i * n + y, (i + 1) * n + x, 0);
		}
	}
	int ans = spfa();
	if (ans == INF)cout << "-1\n";
	else cout << ans << "\n";
	return 0;
}