鏈接: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]]);
}
}