【模板】dij求助
查看原帖
【模板】dij求助
247220
StarryWander楼主2021/4/17 11:39

不是WA就是MLE,RE

求调/kel

#include<bits/stdc++.h>
using namespace std;
const int inf=1e9;
int d[200005];
int w[20005][20005];
bool flag[2000005]; 
int n,m,s,e;
void dijkstra(int s){
	for(int u=1;u<=n;u++) d[u]=inf;
	d[s]=0;
	for(int i=1;i<=n;i++){
		int mn=inf,u=0;
		for(int v=1;v<=n;v++){
			if(!flag[v]&&d[v]<mn){
				mn=d[v],u=v;
			}
		}
		flag[u]=1;
		for(int v=1;v<=n;v++){
			d[v]=min(d[v],d[u]+w[u][v]);
//			if(!flag[v]&&d[u]+w[u][v]<d[v]){
//				d[v]=d[u]+w[u][v];
//			}
		}
	}
}
int main(){
	int s,e;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			w[i][j]=inf;
		}
	}
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		w[x][y]=z;
	}
	dijkstra(1);
	for(int i=1;i<=n;i++){
		if(d[i]!=inf) cout<<d[i]<<" ";
		else cout<<"inf "; 
	}	
	return 0;
}

2021/4/17 11:39
加载中...