堆优化dijkstra

47 篇文章 1 订阅
22 篇文章 0 订阅

题目链接:https://www.luogu.com.cn/problem/CF20C
本题是最短路模板题
spfa的做法

//spfa
#include<iostream>
#include<stack>
#include<queue>
#include<cstring>
using namespace std;
stack<int>st;
struct edge{
 long long n,o,v;
}e[200005];
long long w,last[200005],dist[200005],flag[200005],n,m;
long long yes,pre[200005];
inline void add(long long x,long long y,long long z){
 w++;
 e[w].n=last[x];
 last[x]=w;
 e[w].o=y;
 e[w].v=z;
}

void spfa(long long s){
 long long x,w1,y;
 memset(dist,42,sizeof(dist));
 dist[s]=0;
 memset(flag,0,sizeof(flag));
 flag[s]=1;
 queue<long long>q;
 q.push(s);
 while(q.size()){
  x=q.front();
  if(x==n)yes=1;
  q.pop();
  flag[x]=0;
  for(long long w1=last[x];w1;w1=e[w1].n){
   y=e[w1].o;
   if(dist[x]+e[w1].v<dist[y]){
    dist[y]=dist[x]+e[w1].v;
    pre[y]=x;
    if(flag[y]==0){
     flag[y]=1;
     q.push(y);
    }
   }
  }
 }
}

int main(){
	//freopen("1.in","r",stdin);
	//freopen("1.out","w",stdout);
 cin>>n>>m;
 long long x,y,v;
 for(long long i=1;i<=m;i++){
  scanf("%lld%lld%lld",&x,&y,&v);
  add(x,y,v);
  add(y,x,v);
 }
 spfa(1);
 if(yes==0){
 	cout<<-1<<endl;
 	return 0;
 }
 int now=n;
 while(now){
 	st.push(now);
 	now=pre[now];
 }
 while(st.size()>0){
 	cout<<st.top()<<" ";
 	st.pop();
 }
 return 0;
}

堆优化dijkstra的做法

#include<iostream>
#include<cstring>
#include<queue>
#include<stack>
using namespace std;
#define N 110000
#define M 210000
long long last[N],yes,pre[N],dist[N],x,y,z,n,m,w,flag[N];
struct edge{
	long long n,o,l;
}e[M];
struct arr{
	long long di,id;
};
stack<long long>st;
priority_queue<arr,vector<arr>,less<arr> >q;
bool operator <(arr a,arr b){//结构体小根堆
	return a.di>b.di;
}
void add(long long x,long long y,long long z){
	w++;e[w].n=last[x];last[x]=w;e[w].o=y;e[w].l=z;
}
void dijkstra(long long s){
	long long x,y;
	memset(dist,42,sizeof(dist));//初始化所有距离
	dist[s]=0;
	arr po;
	po.di=0;
	po.id=s;
	q.push(po);//压入起点
	memset(flag,0,sizeof(flag));
	while(q.size()>0){
		x=q.top().id;
		if(x==n)yes=1;//判断是否能到终点
		q.pop();//拿出最近的点
		if(flag[x]==1)continue;//如果这个点已经拿出来过了,则最短的已经更新,当前的此点距离一定不是最短的
		flag[x]=1;//将当前点标记成已经更新过
		for(int w1=last[x];w1;w1=e[w1].n){//搜索它可以更新出去的点
			y=e[w1].o;
			if(dist[x]+e[w1].l<dist[y]){//如果可以更新这个点的答案
				pre[y]=x;//记录下这个点的前导
				dist[y]=dist[x]+e[w1].l;//更新距离
				po.id=y;
				po.di=dist[y];
				q.push(po);//将这个点的最新情况放入小根堆,当前点旧的信息将来会自动淘汰
			}
		}
	}
}
int main(){
	//freopen("1.in","r",stdin);
	//freopen("1.out","w",stdout);
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=m;i++){
		scanf("%lld%lld%lld",&x,&y,&z);
		add(x,y,z);//增加双向边
		add(y,x,z);
	}
	dijkstra(1);//运行dijkstra
	if(yes==0){
		cout<<-1<<endl;
		return 0;
	}
	long long now=n;
	while(now){//用栈记录从终点到起点的情况
		st.push(now);
		now=pre[now];
	}
	while(st.size()>0){//反向输出
		cout<<st.top()<<" ";
		st.pop();
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
C++优化Dijkstra算法是一种用于求解单源最短路问题的算法,它是Dijkstra算法的一种优化。在Dijkstra算法中,我们通过一个数组记录节点到起点的距离,并通过一个集合记录已经确定最短路的节点,然后每次从集合中选择距离起点最近的节点进行松弛操作,直到集合为空为止。但是这种方法的时间复杂度是O(n^2)的,无法处理大规模图的问题。 优化Dijkstra算法则是通过使用优化选择距离起点最近的节点的过程,从而将时间复杂度降低到O(mlogn),其中n为节点数,m为边数。具体实现过程如下: 1. 初始化:将起点加入中,距离为0,其他节点距离为无穷大。 2. 取出顶元素:每次从中取出距离起点最近的节点,将其标记为已经确定最短路。 3. 松弛操作:遍历与该节点相邻的所有节点,如果当前节点到起点的距离可以通过该节点更新,则更新该节点到起点的距离,并将该节点加入中。 4. 重复步骤2和3,直到为空或者顶元素为终点。 下面是C++代码实现: ```c++ #include <bits/stdc++.h> using namespace std; const int N = 1e5 + 5; const int INF = 0x3f3f3f3f; int n, m, s, t; int head[N], cnt = 0; int dis[N], vis[N]; struct Edge { int v, w, nxt; } e[N << 1]; struct Node { int u, d; bool operator < (const Node &rhs) const { return d > rhs.d; } }; void add_edge(int u, int v, int w) { e[++ cnt] = (Edge) {v, w, head[u]}; head[u] = cnt; } void dijkstra() { memset(dis, 0x3f, sizeof(dis)); memset(vis, 0, sizeof(vis)); priority_queue<Node> q; q.push((Node) {s, 0}); dis[s] = 0; while (!q.empty()) { int u = q.top().u; q.pop(); if (vis[u]) continue; vis[u] = 1; for (int i = head[u]; i; i = e[i].nxt) { int v = e[i].v, w = e[i].w; if (dis[v] > dis[u] + w) { dis[v] = dis[u] + w; if (!vis[v]) q.push((Node) {v, dis[v]}); } } } } int main() { cin >> n >> m >> s >> t; for (int i = 1; i <= m; i ++) { int u, v, w; cin >> u >> v >> w; add_edge(u, v, w); add_edge(v, u, w); } dijkstra(); cout << dis[t] << endl; return 0; } ``` 其中,priority_queue<Node> q;表示使用小根来存储节点,Node结构体中的u表示节点编号,d表示该节点到起点的距离。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值