绿叶衬紫花(玄关)
查看原帖
绿叶衬紫花(玄关)
592751
xiaoke2021楼主2024/6/1 13:22

灵异RE,64pts

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int MAXN=100000+5;
const int inf=0x7ffffffffffffff;
struct node{
	int v,w;
	friend bool operator<(node a,node b){
		return a.w>b.w;
	}
};
int n,m,k;
int s,t;
int a,b,c;
int w[MAXN],to[MAXN],nxt[MAXN],head[MAXN];
int cnt;
priority_queue<node> q;
int dis[MAXN];
bool vis[MAXN];
int ans=inf;
void add(int a,int b,int c){
	w[++cnt]=c;
	to[cnt]=b;
	nxt[cnt]=head[a];
	head[a]=cnt;
}
void Dijkstra(){
	for(int i=0;i<MAXN;i++) dis[i]=inf;
	dis[s]=0;
	
	node tmp;
	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=head[u];i;i=nxt[i]){
			if(dis[to[i]]>dis[u]+w[i]){
				dis[to[i]]=dis[u]+w[i];
				tmp.v=to[i],tmp.w=dis[to[i]];
				q.push(tmp);
			}
		}
	}
}
signed main(){
	cin>>n>>m>>k;
	cin>>s>>t;
	for(int i=0;i<m;i++){
		cin>>a>>b>>c;
		add(a,b,c);
		add(b,a,c);
		for(int j=1;j<=k;j++){
			add(a+j*n,b+j*n,c);
			add(b+j*n,a+j*n,c);
			add(a+(j-1)*n,b+j*n,0);
			add(b+(j-1)*n,a+j*n,0);
		}
	}Dijkstra();
	for(int i=0;i<=k;i++) ans=min(ans,dis[t+i*n]);
	cout<<ans;
	return 0;
}
2024/6/1 13:22
加载中...