ZOJ1232 Adventure of Super Mario spfa上的dp

很早之前听说有一种dp是在图上的dp，然后是在跑SPFA的时候进行dp，所以特地找了一题关于在SPFA的时候dp的。

题意：1~a是村庄 a+1~a+b是城堡，存在m条无向边。求由a+b->1的最短路，但是你有很多飞鞋，每双飞鞋可以跑一个固定的距离l,但是跑的时候要是碰到了城堡就要停下来，而且也不能停在路中间。

思路和代码参考了下面的这个网址：http://blog.csdn.net/acm_cxlove/article/details/8679230

思路：正常来说我们就跑SPFA就可以了，每次从队列里取点更新，但因为有了跑鞋，所以每次取点的时候还有多一种转移途径，因此我们就需要知道每次用了跑鞋之后从当前的点能够到达哪些点，这个时候我们就需要两点之间的距离，因为两点之间不能经过城堡，所以在跑SPFA的时候我们对于那些城堡的点我们就不再压入的跑SPFA.然后当dis[u][v]<=l的时候我们就可以转移了。

#pragma warning(disable:4996)
#include<cstring>
#include<cstdio>
#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
#include<queue>
#define maxn 120
#define maxm 120*120
#define inf 0x3f3f3f3f
using namespace std;

int a, b, m, l, k;

struct Edge
{
	int u, v, w;
	Edge(){}
	Edge(int ui, int vi, int wi) :u(ui), v(vi), w(wi){}
}e[maxm];
int ecnt;

int dis[maxn][maxn];
int vis[maxn];
int first[maxn];
int nxt[maxm];

void add(int u, int v, int w)
{
	e[ecnt].u = u; e[ecnt].v = v; e[ecnt].w = w;
	nxt[ecnt] = first[u];
	first[u] = ecnt++;
}

void spfa()
{
	memset(dis, 0x3f, sizeof(dis));
	queue<int> que;
	for (int loc = 1; loc <= a + b; loc++){
		while (!que.empty()) que.pop();
		memset(vis, 0, sizeof(vis));
		que.push(loc); vis[loc] = 1;
		dis[loc][loc] = 0;
		while (!que.empty()){
			int u = que.front(); que.pop(); vis[u] = 0;
			for (int i = first[u]; i != -1; i = nxt[i]){
				int v = e[i].v, w = e[i].w;
				if (dis[loc][v] > dis[loc][u] + w){
					dis[loc][v] = dis[loc][u] + w;
					if (v <= a&&!vis[v]){
						que.push(v); vis[v] = 1;
					}
				}
			}
		}
	}
}

int dp[maxn][15];

int main()
{
	int T; cin >> T;
	while (T--)
	{
		scanf("%d%d%d%d%d", &a, &b, &m, &l, &k);
		memset(first, -1, sizeof(first)); ecnt = 0;
		int ui, vi, wi;
		for (int i = 0; i < m; i++){
			scanf("%d%d%d", &ui, &vi, &wi);
			add(ui, vi, wi);
			add(vi, ui, wi);
		}
		spfa();
		memset(dp, 0x3f, sizeof(dp));
		memset(vis, 0, sizeof(vis));
		queue<int> que;
		que.push(a + b);
		dp[a + b][k] = 0;
		vis[a + b] = 1;
		while (!que.empty()){
			int u = que.front(); que.pop(); vis[u] = 0;
			for (int sh = 0; sh <= k; sh++){
				for (int i = first[u]; i != -1; i = nxt[i]){
					int v = e[i].v, w = e[i].w;
					if (dp[v][sh] > dp[u][sh] + w){
						dp[v][sh] = dp[u][sh] + w;
						if (!vis[v]){
							que.push(v); vis[v] = 1;
						}
					}
				}
				if (!sh) continue;
				for (int v = 1; v <= a + b; v++){
					if (u != v&&dis[u][v] <= l){
						if (dp[v][sh - 1] > dp[u][sh]){
							dp[v][sh - 1] = dp[u][sh];
							if (!vis[v]){
								que.push(v); vis[v] = 1;
							}
						}
					}
				}
			}
		}
		int ans = inf;
		for (int i = 0; i <= k; i++){
			ans = min(ans, dp[1][i]);
		}
		printf("%d
", ans);
	}
	return 0;
}

---恢复内容结束---