样例没过
查看原帖
样例没过
550957
Anonymely楼主2022/7/11 19:57
#include<bits/stdc++.h>
using namespace std;

const int N=100005;
#define pii pair<int,int>

int n,m,k,s,t;

struct node{
	int to,next,del;
}e[N*15];

int head[N],cnt,dis[N],vis[N];

void add(int u,int v,int k=0){
	e[++cnt].to=v;
	e[cnt].next=head[u];
	e[cnt].del=k;
	head[u]=cnt;
}

void dijkstra(int s){
	priority_queue<pii,vector<pii>,greater<pii> > q;
	memset(dis,0x3f,sizeof(dis));
	q.push({0,s});
	dis[s]=0;
	while(!q.empty()){
		int x=q.top().second;
		q.pop();
		if(vis[x])continue;
		vis[x]=1;
		for(int i=head[x];i;i=e[i].next){
			int v=e[i].to,k=e[i].del;
			if(dis[v]>dis[x]+k){
				dis[v]=dis[x]+k;
				q.push({dis[v],v});
			}
		}
	}
}

signed main(){
	cin>>n>>m>>k>>s>>t;
	for(int i=1;i<=n;i++){
		int u,v,d;cin>>u>>v>>d;
		add(u,v,d);
		add(v,u,d);
		for(int j=1;j<=k;j++){
			add(u+(j-1)*n,v+j*n);
			add(v+(j-1)*n,u+j*n);
			add(u+j*n,v+j*n,d);
			add(v+j*n,u+j*n,d);
		}
	}
	for(int i=1;i<=k;i++)add(t+(i-1)*n,t+i*n); 
	dijkstra(s);
	cout<<dis[t+k*n];
}

样例没过,c

2022/7/11 19:57
加载中...