链接:http://acm.hdu.edu.cn/showproblem.php?pid=6705
题意:求一个有向有权图中,第k短的路径。
思路:用小顶堆维护路径权值,每次扩展从上一个点到当前点的下一条边和由当前点出去的最小的一条边即可。
官方题解:先把每条边以 形式放进堆,堆按路径权值从小到大排序,然后每次取出堆顶,用v的出边扩展
新的路径。但是一个点的出度可能会非常大(如菊花图),可以发现,将出边排序之后,每次只需要扩
展当前点最小的出边,和扩展到当前点的边的下一条边即可。堆中需要记录当前结点,当前距离,上一
节点距离,扩展到当前节点时下一条应该扩展的边。(注意,如果一次性扩展当前点连出去的所有权值
相同的边,是会TLE的,实际上也是没有必要的。)
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 5e4+10;
struct node
{
int to;
ll w;
node(){}
node(int to,ll w):to(to),w(w){}
};
struct Node
{
int pre,u,cur;
ll predis,dis;
Node(){}
Node(int ppre,int uu,int ccur,ll ppredis,ll ddis)
{
pre=ppre,u=uu;
cur=ccur,predis=ppredis;
dis=ddis;
}
friend bool operator <(const Node& a,const Node& b)
{
return a.dis>b.dis;
}
}now;
vector <node> g[N];
int n,m,q,maxk,que[N];
ll ans[N];
void Init()
{
maxk=0;
for(int i=1;i<=n;i++)
g[i].clear();
}
bool cmp(const node& a,const node& b)
{
return a.w<b.w;
}
int main(void)
{
int t,u,v,cnt;
ll w;
scanf("%d",&t);
while(t--)
{
Init();
cnt=0;
scanf("%d%d%d",&n,&m,&q);
priority_queue<Node> qu;
for(int i=1;i<=m;i++)
{
scanf("%d%d%lld",&u,&v,&w);
g[u].push_back(node(v,w));
}
for(int i=1;i<=n;i++)
{
sort(g[i].begin(),g[i].end(),cmp);
if(g[i].size())
qu.push(Node(i,g[i][0].to,0,0,g[i][0].w));
}
for(int i=1;i<=q;i++)
{
scanf("%d",&que[i]);
maxk=max(maxk,que[i]);
}
while(!qu.empty())
{
now=qu.top();
qu.pop();
//cout<<now.dis<<endl;
ans[++cnt]=now.dis;
if(cnt==maxk) break;
if(now.cur+1<g[now.pre].size())
qu.push(Node(now.pre,g[now.pre][now.cur+1].to,now.cur+1,now.predis,now.predis+g[now.pre][now.cur+1].w));
if(g[now.u].size())
qu.push(Node(now.u,g[now.u][0].to,0,now.dis,now.dis+g[now.u][0].w));
}
for(int i=1;i<=q;i++)
printf("%lld\n",ans[que[i]]);
}
}