题意:给定一个n(1 <= n <= 1000)个点,m(1 <= m <= 100000)条边的有向图,求S到T的第k短路。
时间限制:4000ms
空间限制:64M
通过时间:219ms
通过空间:9132K
分析:这道题瓶颈不在于时间,而在于空间,朴素算法不外乎BFS,第k次访问到终点的时候,就是答案,容易理解,但是状态量太多,空间只有仅仅的64M,放不下,我们需要寻找一个的更好的搜索方式。
于是我们想到了传说中的启发式搜索A*,网上找到了一篇很好的文章:http://www.cnblogs.com/technology/archive/2011/05/26/2058842.html
根据这篇文章,我们来设计乐观估价函数,不外乎是从终点反向走一遍得出每个点的最短路,作为H,然后已经走过的路程作为G,F = G + H.
这里采用了邻接表建图,需要注意的是,如果hd初始是0,可以选择都赋值成-1,也可以选择边序号从1开始(也就是先进行++cnt)
之后跑SPFA,模板,没啥说的。
然后进行A*搜索过程,用一个结构体A,来存放to结点的F和G.
搜索的时候需要注意以下几个事项:
1.在放第一个结点之前,得先看看是不是能到,不能到直接返回-1.
2.特判S == T的情况,当S == T时,k++.
3.每次取F值最小的元素,所以要用priority_queue小根堆,因为优先队列默认是大根堆,所以重载小于运算符的时候定义成大于符。
以下为扯淡内容:
这道题当时做的时候是几近崩溃的,明明跟题解基本上一模一样,但死活MLE,在进行了一页半(大概30来次)的提交后,终于发现我SPFA的源点用的是n,不是T!!!!瞬间愤怒了,望着8楼的窗外,觉得人生无比美好...
#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;
const int inf = 0x3f3f3f3f, maxn = 1005, maxm = 100005;
bool inq[maxn];
int n, x, y, z, s, t, k, m, cnt, d[maxn], hd[maxn], hd2[maxn];
struct Edge {
int to, w, nxt;
}edge[maxm], edge2[maxm];
void add(int x, int y, int z) {
edge[cnt].w = edge2[cnt].w = z;
edge[cnt].nxt = hd[x];
edge2[cnt].to = x;
edge2[cnt].nxt = hd2[y];
hd[x] = hd2[y] = cnt;
}
void spfa(int s) {
queue<int> q;
memset(d, inf, sizeof d);
d[s] = 0;
q.push(s);
while(!q.empty()) {
int u = q.front(); q.pop();
inq[u] = 0;
for(int i = hd[u]; i; i = edge[i].nxt) {
Edge &e = edge[i];
if(d[e.to] > d[u] + e.w) {
d[e.to] = d[u] + e.w;
if(!inq[e.to]) inq[e.to] = 1, q.push(e.to);
}
}
}
}
struct A {
int to, g, f;
bool operator < (const A &rhs) const {
return f > rhs.f;
}
};
int Asrch() {
if(d[s] == inf) return -1;
if(s == t) k++;
priority_queue<A> q;
A e, ne;
e.to = s, e.g = 0, e.f = d[s];
q.push(e);
while(!q.empty()) {
e = q.top(); q.pop();
if(e.to == t) k--;
if(k == 0) return e.g;
for(int i = hd2[e.to]; i; i = edge2[i].nxt) {
ne.to = edge2[i].to;
ne.g = e.g + edge2[i].w;
ne.f = ne.g + d[ne.to];
q.push(ne);
}
}
return -1;
}
int main() {
scanf("%d%d", &n, &m);
while(m--) scanf("%d%d%d", &x, &y, &z), add(y, x, z);
scanf("%d%d%d", &s, &t, &k);
spfa(t);
printf("%d\n", Asrch());
return 0;
}