求单源最短路径及距离——dijkstra模板

关于dijkstra的介绍,可以阅读这篇文章
算法之迪杰斯特拉(dijkstra)非常详细介绍

在这里使用<后继,权值>的结构体edge来构建邻接表G。dis表示源点到各点的距离。 题中结点编号为 1~n,为方便这里以 0~n-1 的编号进行处理,再以1~n的编号输出


堆优化dijkstra求最短路径的距离(若不可达,输出URA,即 2 31 − 1 2^{31}-1 2311

思想:因为dijkstra有个不断选出源点到未访问点最短的路径的过程,所以可以使用堆(priority_queue)对距离进行排序,这个距离是通过构建pair<距离,后继>实现的,堆会对pair.first进行排序。值得注意的是,选取的堆顶距离若不是最小值,说明入队之后该点又经过最小值的更新,堆顶这个值不是当前的最小距离,所以舍弃这个值。这种方法不需要辅助数组vis判断是否访问过。
使用堆优化后,复杂度为 O ( ∣ E ∣ l o g ∣ V ∣ ) O(|E|log|V|) O(ElogV)

例题:
洛谷-【模板】单源最短路径(弱化版)
洛谷-【模板】单源最短路径(标准版)

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
const int MAX_N=1e4+10;
const int INF=1e9+10;
const int URA=(2<<30)-1;
typedef pair<int,int> pii;
struct edge
{
	int to,cost;	
	edge(int t,int c):to(t),cost(c){}
};
int n,m,s;
int dis[MAX_N];
vector<edge> G[MAX_N];
void dijkstra(int s)//s:源点
{
	priority_queue<pii,vector<pii>,greater<pii>>q;
	for(int i=0;i<n;i++)
		dis[i]=INF;
	dis[s]=0;
	q.push(pii(0,s));
	while(!q.empty())
	{
		pii p=q.top();
		q.pop();
		int u=p.second;
		if(dis[u]<p.first)//若堆顶的距离不是最小值,说明入队之后又经过最小值的更新
			continue;
		for(int i=0;i<G[u].size();i++)
		{
			int v=G[u][i].to;
			if(dis[u]+G[u][i].cost<dis[v])
			{
				dis[v]=dis[u]+G[u][i].cost;
				q.push(pii(dis[v],v));
			}
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>s;
	int u,v,w;
	while(m--)
	{
		cin>>u>>v>>w;
		G[u-1].push_back(edge(v-1,w));
	}
	dijkstra(s-1);
	for(int i=0;i<n;i++)
		cout<<(dis[i]==INF?URA:dis[i])<<" ";
	return 0;
}

求其中一条1到n的最短路径(若不可达,仅输出-1)
思想:在原基础上设置前趋数组pre,当某个点更新最短距离时,使用pre保存其前趋结点,dijkstra运算结束后,可以从结点n往上回溯,由于回溯的顺序和输出顺序相反,所以使用栈stack保存和输出。

例题:codeforces-Dijkstra?

#include<iostream>
#include<vector>
#include<queue>
#include<stack>
using namespace std;
typedef long long ll;
typedef pair<ll,int> pli;
const ll INF=9e13+7;
const int MAX_N=1e5+10;
struct edge
{
	int to;
	ll cost;	
	edge(int t,ll c):to(t),cost(c){}
};
int n,m;
int pre[MAX_N];//记录每个点的前驱结点
ll dis[MAX_N];
vector<edge> G[MAX_N];//邻接表
void dijkstra(int s)
{
	priority_queue<pli,vector<pli>,greater<pli> > q;
	for(int i=0;i<n;i++)
	{
		dis[i]=INF;
		pre[i]=-1;
	}
	dis[s]=0;
	q.push(pli(0,s));
	while(!q.empty())
	{
		pli p=q.top();
		q.pop();
		int u=p.second;
		if(p.first>dis[u])
			continue;
		for(int i=0;i<G[u].size();i++)
		{
			int v=G[u][i].to;
			if(dis[v]>G[u][i].cost+dis[u])
			{
				dis[v]=G[u][i].cost+dis[u];
				q.push(pli(dis[v],v));
				pre[v]=u;//记录前趋
			}
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	int u,v,w;
	while(m--)
	{
		cin>>u>>v>>w;
		G[u-1].push_back(edge(v-1,w));
		G[v-1].push_back(edge(u-1,w));
	}
	dijkstra(0);
	if(dis[n-1]==INF)
		cout<<-1<<"\n";
	else
	{
		int i=n-1;
		stack<int> s;
		while(i!=-1)
		{
			s.push(i);
			i=pre[i];
		}
		while(!s.empty())
		{
			cout<<s.top()+1<<" ";
			s.pop();
		}
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值