hdu 6705 path 堆

求第k短的路径。

每一个路径，必定是由一个路径加一条边构成。而且新得到的路径长度一定长于原先的路径长度。

所以我们最开始把所有出边按长短排序。只要把每个点出发的最短的边加入一个堆。然后每次选出堆里最短的边。pot,pos,len。表示这个路径最后一条边是从pot走了其第pos条出边，到某个点x，路径总长为len。

然后产生两个新的路径。一个还是从pot点走，走pos+1号边。一个是从x点走，走0号边。

#include <cstdio>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int MAXN = 50010;
struct edg
{
    int to,val;
    edg (int _to,int _val):to(_to),val(_val){}
    friend bool operator < (edg a,edg b)
    {
        return a.val < b.val;
    }
};
struct cas
{
    int pot,pos;
    ll len;
    cas (int _pot,int _pos,ll _len):pot(_pot),pos(_pos),len(_len){}
    friend bool operator < (cas a,cas b)
    {
        return a.len > b.len;
    }
};
priority_queue <cas> que;
vector <edg> vec[MAXN];
int T,n,m,q,tot,maxn,k[MAXN];
ll ans[MAXN];
int main()
{
    for (scanf("%d",&T);T != 0;T--)
    {
        for (int i = 1;i <= n;i++)
            vec[i].clear();
        while (que.empty() == false)
            que.pop(); 
        maxn = tot = 0;
        scanf("%d%d%d",&n,&m,&q);
        int tx,ty,tv;
        for (int i = 1;i <= m;i++)
        {
            scanf("%d%d%d",&tx,&ty,&tv);
            vec[tx].push_back(edg(ty,tv));
        }
        for (int i = 1;i <= q;i++)
        {
            scanf("%d",&k[i]);
            maxn = max(maxn,k[i]);
        }
        for (int i = 1;i <= n;i++)
            sort(vec[i].begin(),vec[i].end());
        for (int i = 1;i <= n;i++)
            if (vec[i].size() != 0)
                que.push(cas(i,0,vec[i][0].val));
        while (que.empty() == false)
        {
            cas tp = que.top();
            que.pop();
            if (tot >= maxn)
                break;
            ans[++tot] = tp.len;
            if (tp.pos + 1 < vec[tp.pot].size())
                que.push(cas(tp.pot,tp.pos + 1,tp.len - vec[tp.pot][tp.pos].val + vec[tp.pot][tp.pos + 1].val));
            if (vec[vec[tp.pot][tp.pos].to].size() != 0)
                que.push(cas(vec[tp.pot][tp.pos].to,0,tp.len + vec[vec[tp.pot][tp.pos].to][0].val));
        }
        for (int i = 1;i <= q;i++)
            printf("%lld\n",ans[k[i]]);
    }
    return 0;
}