链接: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;
}