玄关
查看原帖
玄关
941628
Max_robot楼主2024/9/2 18:23
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e7+10;
long long n, m, k;
long long s, t;
long long dis[v], h[maxn], to[maxn], val[maxn], nxt[maxn];
bool vis[maxn];
long long cnt;
long long ans=LONG_LONG_MAX;
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 a, int b, int c){
	to[++cnt]=b;
	val[cnt]=c;
	nxt[cnt]=h[a];
	h[a]=cnt;
}
void Dijkstra(){
	for(int i=1;i<=n;i++)
		dis[i]=LONG_LONG_MAX;
	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]=true;
		for(int i=h[u];i;i=nxt[i]){
			if(dis[to[i]]>dis[u]+val[i]){
				dis[to[i]]=dis[u]+val[i];
				tmp.w=dis[to[i]], tmp.v=to[i];
				q.push(tmp);
			}
		}
	}
}
int main(){
	cin>>n>>m>>k;
	cin>>s>>t;
	int u, v, w;
	for(int i=0;i<m;i++){
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
		for(int j=1;j<=k;j++){
			add(u+j*n,v+j*n,w);
			add(v+j*n,u+j*n,w);
			add(u+j*n-n,v+j*n,0);
			add(v+j*n-n,u+j*n,0);
		}
	}
	Dijkstra();
	for(int j=0;j<=k;j++)
		ans=min(ans,dis[t+j*n]);
	cout<<ans<<endl;
	return 0;
}

Dijkstra读迪杰斯特拉迪杰斯特拉

2024/9/2 18:23
加载中...