POJ 2449 K短路

点击就送屠龙宝刀

题目大意
给定n个点，m条边构成的一个图
求s到t的第k短路
输入描述
n m
m行边（单向）
s,t,k
输出描述
如题意

调了好久，各种神奇的bug
果然A*这种东西……
有个最短路算法叫迪杰斯特拉
第一次到某点的时候就是最短路
那么猜想：第k次到达的时候是不是就是k短路呢
显然的（我不会证明）
然后
加个A*的h()表示到终点的距离（反向建图，然后跑最短路）
那么
就行了

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 100000 + 5;
const int MAXM = 100000 + 5;
struct edge
{
    int f,t,v;
}l[MAXM << 1];
int n,m;
int dis[MAXN];
struct zt
{
    int u,v;
    bool operator < (const zt b)const
    {
        return v > b.v;
    }
};
int first[MAXN],next[MAXM << 1],tot;
void init()
{
    memset(first,0xfff,sizeof(first));
    tot = 0;
    return;
}
void build(int f,int t,int v)
{
    l[++tot] = (edge){f,t,v};
    next[tot] = first[f];
    first[f] = tot;
    return;
}
int done[MAXN];
priority_queue <zt> q;
int s,e;
void dij()
{
    memset(dis,0x3f,sizeof(dis));
    dis[e] = 0;
    q.push((zt){e,0});
    while(!q.empty())
    {
        zt x = q.top();
        int u = x.u;
        q.pop();
        if(done[u])
            continue;
        //if(u == s)
        //  return;与最短路的不同，虽然这是init()
        done[u] = true;
        for(int i = first[u];i != -1;i = next[i])
        {
            int v = l[i].t;
            if(dis[v] > dis[u] + l[i].v)
            {
                dis[v] = dis[u] + l[i].v;
                q.push((zt){v,dis[v]});
            }
        }
    }
    return;
}
int k;
int Astar()
{
    while(!q.empty())
        q.pop();
    if(s == e)
        k++;
    q.push((zt){s,dis[s]});
    while(!q.empty())
    {
        zt x = q.top();
        int u = x.u;
        q.pop();
        if(u == e)
            if(!--k)
                return x.v;
        for(int i = first[u];i != -1;i = next[i])
        {
            int v = l[i].t;
            q.push((zt){v,x.v - dis[u] + dis[v] + l[i].v});
        }
    }
    return -1;
}
int f[MAXN],t[MAXN],v[MAXN];
int main()
{
    init();
    scanf("%d %d",&n,&m);
    for(int i = 1;i <= m;i ++)
    {
        scanf("%d %d %d",&f[i],&t[i],&v[i]);
        build(t[i],f[i],v[i]);
    }
    scanf("%d %d %d",&s,&e,&k);
    dij();
    init();
    for(int i = 1;i <= m;i ++)
        build(f[i],t[i],v[i]);
    printf("%d\n",Astar());
    return 0;
}