优先队列优化的Dijkstra,70分求助
查看原帖
优先队列优化的Dijkstra,70分求助
579489
Vigilant_Yaksha楼主2023/2/1 08:01
#include<bits/stdc++.h>
#define maxn 100010
using namespace std;
int n,m,s,cnt;
int to[maxn*5],nex[maxn*5],head[maxn*5],dis[maxn*5],num[maxn];
bool vis[maxn];
const int inf=100000; 
struct node{
	int v,w;
	friend bool operator <(node a,node b){
		return a.w > b.w;
	}
}tmp;
priority_queue<node>q;
void add(int x,int y,int z){
	to[++cnt]=y;
	nex[cnt]=head[x];
	head[x]=cnt;
	num[cnt]=z;
}
void dijkstra(){
	for(int i=1;i<=n;i++)
		dis[i]=inf;
	dis[s]=0;
	tmp.v=s;
	tmp.w=0;
	q.push(tmp);
	while(!q.empty()){
		int u=q.top().v;
		q.pop();
		if(vis[u])continue;
		vis[u]=1;
		for(int i=head[u];i;i=nex[i])
			if(dis[to[i]]>=(long long)(dis[u]+num[i])){
				dis[to[i]]=dis[u]+num[i];
				tmp.w=dis[to[i]],tmp.v=to[i];
				q.push(tmp); 
			}
	} 
}
int main(){
	cin>>n>>m>>s;
	for(int i=1,a,b,c;i<=m;i++){
		cin>>a>>b>>c;
		add(a,b,c);
	}
	dijkstra();
	for(int i=1;i<=n;i++){
		if(dis[i]==100000)
			cout<<2147483647<<" ";
		else
			cout<<dis[i]<<" ";
	}
	return 0;
}
2023/2/1 08:01
加载中...