模板-------单源最短路之djkjstra(堆优化)

注:因为垃圾博主重载运算符用的很烂,所以这里并没用结构体

#include<iostream>
#include<cstdio>
#include<queue>
#define MP make_pair
#define se second 
#define fi first
#define LL long long
using namespace std;
typedef pair<int,int> pll;
const int MAXN=5e5+5;
int n,m,s;
int dis[MAXN];
int h[MAXN],to[MAXN],nxt[MAXN],len[MAXN],num_edge; 
priority_queue<pll,vector<pll>,greater<pll> >que;
inline void add_edge(int x,int y,int z)
{
	nxt[++num_edge]=h[x];
	to[num_edge]=y;
	len[num_edge]=z;
	h[x]=num_edge;
}
int main()
{
	int U,V,W;
	scanf("%d%d%d",&n,&m,&s);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&U,&V,&W);
		add_edge(U,V,W);
	}
	for(int i=1;i<=n;i++)
	 dis[i]=2147483647;
	dis[s]=0;
	que.push(MP(dis[s],s));
	while(!que.empty())
	{
		int x=que.top().se;
		if(que.top().fi!=dis[x])
		{
		 que.pop();continue;
		} 
		que.pop(); 
		for(int i=h[x];i;i=nxt[i])
		{
			int k=to[i],t=len[i];
			if(dis[k]>dis[x]+t)
			{
				dis[k]=dis[x]+t;
				que.push(MP(dis[k],k));
			}
		}
	}
	for(int i=1;i<=n;i++)
	 printf("%d ",dis[i]);
	printf("\n");
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_42920131/article/details/85368443