求调(悬赏1~3关)
查看原帖
求调(悬赏1~3关)
385952
f2_longtian楼主2023/2/6 21:44

RT,加了堆优化的dijkstra

#include<bits/stdc++.h>
using namespace std;
struct edge{
	int to,l;
};
typedef pair<int,int> PII;
vector<edge> v[1005];
bool vis[1005],dis[1005];
int n,m,s,e,x,y,z;
priority_queue<PII,vector<PII>,greater<PII> > pq;
void dijkstra(int x){
	memset(dis,127,sizeof(dis));
	dis[x]=0;
	pq.push({0,x});
	while(!pq.empty()){
		PII t=pq.top();
		pq.pop();
		int id=t.second;
		int d=t.first;
		if(vis[id]==true)continue;
		vis[id]=true;
		for(int i=0;i<v[id].size();i++){
			edge y=v[id][i];
			int to=y.to;
			int len=y.l;
			if(dis[to]>d+len){
				dis[to]=d+len;
				pq.push({dis[to],to});
			}
		}
	}
} 
int main(){
	cin>>n>>m>>s;
	for(int i=1;i<=m;i++){
		cin>>x>>y>>z;
		v[x].push_back({y,z});
		v[y].push_back({x,z});
	}
	dijkstra(s);
	for(int i=1;i<=n;i++)cout<<dis[i]<<' ';
	return 0;
}

2023/2/6 21:44
加载中...