哪位大佬帮我看一下这个分层图最短路 P4568
查看原帖
哪位大佬帮我看一下这个分层图最短路 P4568
533297
Patronus楼主2023/7/7 15:02
#include<bits/stdc++.h>
using namespace std;
const int N=60005,M=500005;
int head[N],ver[M],Next[M],edge[N],d[N];
long long n,m,k,s,t,tot;
bool v[N];
priority_queue<pair<int,int > >q;
void add(int x,int y,int z){
	ver[++tot]=y;
	edge[tot]=z;
	Next[tot]=head[x];
	head[x]=tot;
}
void dijkstra(int s){
	memset(d,0x3f,sizeof(d));
	memset(v,0,sizeof(v));
	d[s]=0;
	q.push(make_pair(0,s));
	while(q.size()){
		int x=q.top().second;
		q.pop();
		if(v[x]) continue;
		v[x]=1;
		for(int i=head[x];i;i=Next[i]){
			int y=ver[i],z=edge[i];
			if(d[y]>d[x]+z){
				d[y]=d[x]+z;
				q.push(make_pair(-d[y],y));
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>k;
	cin>>s>>t; 
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		for(int j=0;j<=k;j++){
			add(x+j*n,y+j*n,z);
			add(y+j*n,x+j*n,z);
			if(j!=k){
				add(x+j*n,y+(j+1)*n,0);
				add(y+j*n,x+(j+1)*n,0);
			}
		}
	}
	dijkstra(s);
	int ans=(1<<30);
	for(int i=0;i<=k;i++){
		ans=min(ans,d[t+i*n]);
	}
	
	cout<<ans<<endl;
	return 0;
}
2023/7/7 15:02
加载中...