代码求调,奖励1关注
查看原帖
代码求调,奖励1关注
449051
cccyyyxxx楼主2023/6/19 21:20
#include<bits/stdc++.h>
using namespace std;
const int N = 5e4+5;
inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    return x*f;
}
int n,m,k,s,t,ans=0x7fffffff;
int dis[N],head[N],net[N],val[N],to[N],vis[N],cnt;
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 u,int v,int w){
	to[++cnt]=v;
	net[cnt]=head[u];
	head[u]=cnt;
	val[cnt]=w;
}
void dijkstra(){
	memset(dis,0x3f,sizeof(dis));
	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[i];i;i=net[i]){
			if(dis[to[i]]>dis[u]+val[i]){
				dis[to[i]]=dis[u]+val[i];
				tmp.v=to[i];
				tmp.w=dis[to[i]];
				if(!vis[to[i]])
					q.push(tmp);
			}
		}
	}
}
signed main(){
	n=read(); m=read(); k=read(); s=read(); t=read();
	++s,++t;
	for(int i=1;i<=m;i++){
		int u,v,w;
		u=read(); v=read(); w=read();
		++u,++v;
		add(u,v,w); 
		add(v,u,w);
		for(int j=1;j<=k;j++){
			add(u+(j-1)*n,v+j*n,0);
			add(v+(j-1)*n,u+j*n,0);
			add(u+j*n,v+j*n,w);
			add(v+j*n,u+j*n,w);
		}
	}
	dijkstra();
	for(int i=0;i<=k;i++)
		ans=min(ans,dis[t+i*n]);
	cout<<ans<<endl;
	return 0;
}

2023/6/19 21:20
加载中...