POJ 2449 Remmarguts' Date (第k短路)

题目类型  第k短路

题目意思

给出 n 个点 m 条有向边(n <= 1000, m <= 100000) 问从 s->t的第 k 短路是多少 (点可以重复走)

解题方法

使用A*算法　->  对于 前K短路径问题 和 A*算法 的一些小小总结

具体为

先预处理出所有点到 t 的最短路(即最短时间)　假设为 d[i]

那么建一个如下的优先队列　先把 Node(s, 0) 放进队列中　然后取队列首元素f(取出后要把f出队),　把与f相连的　点v与 f.len+road[f][v] 

即 Node(v, f.len+road[f][v])　放进队列中 (road[f][v] 表示 f->v这条路径的长度)  

不断进行这样的操作当 f.u == t的时候　tk++, 当 tk == k 的时候即找到了第 k 短路

priority<Node>q;

struct Node {

  int u, len; // 表示从s到结点 u 已走了 len, 那么至少还需要走的路程即 d[u]

  bool operator < (const Node & rhs) const { // 则比较函数这样写 len+d[u] > rhs.len+d[rhs.u]　的话则优先级越小
    return len + d[u] > rhs.len + d[rhs.u];
  }

};

注意s == t　最短路不是 0, 即第 i 短路 (1 <= i <= k) 　必须包含有路径

参考代码 - 有疑问的地方在下方留言 看到会尽快回复的

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>

using namespace std;

const int maxn = 1000 + 10;
const int INF = 1<<29;

vector<int>E[maxn], W[maxn], Ek[maxn], Wk[maxn];
int d[maxn];
bool vis[maxn];

struct Node {
	int u, len;
	Node(int tu, int tlen) : u(tu), len(tlen) {}
	Node(){}
	bool operator < (const Node & rhs) const {
		return len + d[u] > rhs.len + d[rhs.u];
	}
};


int k_road(int s, int t, int n, int k) {　// 用A*算法求第 k 短路
	priority_queue<Node>q;
	q.push(Node(s, 0));
	int tk = 0;
	while(!q.empty()) {
		Node f = q.top(); q.pop();
		if(f.u == t) {
			tk++;
			if(tk == k) {
				return f.len;
			}
		}
		int u = f.u;
		for( int i=0; i<Ek[u].size(); i++ ) {
			int v = Ek[u][i];
			q.push(Node(v, f.len+Wk[u][i]));
		}
	}
	return -1;
}

void spfa(int s, int t, int n) {　// 求出其他点到　t 的最短路
	for( int i=1; i<=n; i++ ) d[i] = INF;
	d[t] = 0;
	queue<int>q;
	q.push(t);
	memset(vis, 0, sizeof(vis));
	vis[t] = true;
	while(!q.empty()) {
		int f = q.front(); q.pop(); vis[f] = false;
		for( int i=0; i<E[f].size(); i++ ) {
			int v = E[f][i];
			if(d[v] > d[f] + W[f][i]) {
				d[v] = d[f] + W[f][i];
				if(vis[v] == false) {
					vis[v] = true;
					q.push(v);
				}
			}
		}
	}
}

int main() {
	freopen("in","r", stdin);
	int n, m;
	while(scanf("%d%d", &n, &m) != EOF) {
		int s, t, k;
		for( int i=1; i<=n; i++ ) E[i].clear(), W[i].clear(), Ek[i].clear(), Wk[i].clear();
		for( int i=0; i<m; i++ ) {
			int u, v, w;
			scanf("%d%d%d", &v, &u, &w);
			E[u].push_back(v);
			W[u].push_back(w);　//E和W保存的边是原来的边的反向边　这样就可以从 t 出发更新其他的点
			Ek[v].push_back(u); //Ek和Wk保存的是原来的边
			Wk[v].push_back(w);
		}
		scanf("%d%d%d", &s, &t, &k);
		if(s == t) k++;
		spfa(s, t, n); //预处理最短路
		if(d[s] == INF) {
			printf("-1\n");
			continue;
		}
		/*for( int i=1; i<=n; i++ ) printf("%d ", d[i]);
		printf("\n");*/
		int ans = k_road(s, t, n, k);
		if(ans == -1) printf("-1\n");
		else printf("%d\n", ans);
	}
	return 0;
}