最短路:dijkstra堆优化

24 篇文章 12 订阅
8 篇文章 1 订阅

上集回顾:最短路:dijkstra算法_物联网土猫的博客-CSDN博客

那么按照写文章的尿素,接下来就是我们的优化时间;

纵观代码,我们可以找到一些地方优化,比如:我们能不能在循环上缩减,减少遍历次数呢?答案当然是可以的!

看看代码,大循环肯定不能减,小循环呢?最后一步判断最短路比较肯定不能删掉,那么我们理所当然的应该改动寻找最小权值那里。

这里的最小权值只是一个值,我们需要的是最小权值这个点,所以这里就有了一种、堆优化方案,具体是,第一次循环我们找到最短路,接下来把出边更新之后,我们把所有出边加入我们的堆,这里我们的堆是priority_queue,他实现了堆内部自动排序,但是有个问题是他是大根堆,也就是从大到小排列,我们这里存入权值的时候存入负值也就直接变成小根堆了。

这样的话我们每次只需要取出第一个值,接着遍历,加入即可;

这里的代码我们定义了自动排序的小跟堆,也就不需要存入负值了;

#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N=10010;
int n,m;
int head[N],w[N],e[N],ne[N],idx; 
int dist[N];
bool st[N];
 
typedef pair<int,int> PII;
 
void add(int a,int b,int c) 
{
	e[idx]=b,ne[idx]=head[a],w[idx]=c,head[a]=idx++;
}
 
int dijkstra()
{
	memset(dist,0X3f3f3f3f,sizeof(dist));
	dist[1]=0;
	
	priority_queue<PII ,vector<PII>, greater<PII>> heap;//定义大根堆 
	heap.push({0,1});//压入 
	while(heap.size())
	{
		auto t=heap.top();//自动识别类型 
		heap.pop();
		
		int ver=t.second, distence=t.first;
		if(st[ver]) continue; //如果已经被找过最短路了,就不需要再加入一次,否则会无限循环 
		
		for(int i=head[ver];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>distence+w[i]) 
			{
				dist[j]=distence+w[i];//常规操作 W
				heap.push({dist[j],j});
			} //加入操作;
		}
	}
	
	if(dist[n]==0X3f3f3f3f) return -1;
	else return dist[n];
}
 
int main()
{
	memset(head,-1,sizeof(head));
	cin>>n>>m;
	while(m--)
	{
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
	}
	int t=dijkstra();
	cout<<t<<endl;
	return 0;
}

具体代码:

  • 7
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值