HDU-6705 path(贪心+思维)

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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值