dijkstra堆优化模板2.0

gentle coder

于 2023-07-02 19:24:02 发布

阅读量42

点赞数 1

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/m0_74969835/article/details/131504212

版权

该代码实现了一个Dijkstra算法的C++版本，用于在一个包含加权边的图中寻找从源节点到所有其他节点的最短路径。程序首先读取图的节点数和边信息，然后应用Dijkstra算法计算最短路径，并输出结果。如果路径不存在，则输出-1。

摘要由CSDN通过智能技术生成

#include<bits/stdc++.h>
using namespace std;

#define N 300005
#define inf 1e17

struct node
{
	long long v,w;
};
int n,m,vis[N];
long long d[N];
vector<node>e[N];

void dijkstra()
{
	priority_queue<pair<int,int> >q;
	for(int i=1;i<=n;i++)d[i]=inf;
	d[1]=0;
	q.push({0,1});
	while(q.size())
	{
		pair<int,int>s=q.top();
		q.pop();
		int u=s.second;
		if(vis[u])continue;
		vis[u]=1;
		for(int i=0;i<e[u].size();i++)
		{
			int v=e[u][i].v,w=e[u][i].w;
			if(d[v]>d[u]+w)
			{
				d[v]=d[u]+w;
				q.push({-d[v],v});
			}
		}
	}
}

int main()
{
	int a,b,c;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		cin>>a>>b>>c;
		if(a==b)continue;
		e[a].push_back({b,c});
	}
	dijkstra();
	for(int i=1;i<=n;i++)
	{
		if(d[i]!=inf)cout<<d[i]<<" ";
		else cout<<-1<<" ";
	}
	return 0;
}