90pts WA#11 分层图
查看原帖
90pts WA#11 分层图
667558
_Kamisato_Ayaka_楼主2023/5/2 22:29
#include<bits/stdc++.h>
using namespace std;
const int N = 5e6;
const int inf = 0x7f7f7f7f;
int n,m,k,p=1,ans=1e9;
int s,t;
int head[N],dis[N],vis[N];
struct node{
	int v,w,nxt;
}e[N];
void add(int u,int v,int w){
	e[p].v=v;
	e[p].w=w;
	e[p].nxt=head[u];
	head[u]=p++;
}
void dijkstra(int s){
	typedef pair<int,int> pii;
	priority_queue<pii,vector< pii >,greater< pii > > q;
	dis[s]=0;
	q.push(make_pair(dis[s],s));
	while(!q.empty()){
		int u=q.top().second;
		q.pop();
		if(vis[u]) 
			continue;
		vis[u]=1;
		for(int i=head[u];i!=-1;i=e[i].nxt){
			int v=e[i].v;
			if(dis[v]>dis[u]+e[i].w){
				dis[v]=dis[u]+e[i].w;
				q.push(make_pair(dis[v],v));
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>k;
	cin>>s>>t;
	memset(head,-1,sizeof head);
	memset(dis,inf,sizeof dis);
	for(int i=1;i<=m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
		for(int j=1;j<=k;j++){
			add(j*n+u,j*n+v,w);
			add(j*n+v,j*n+u,w);
			add((j-1)*n+u,j*n+v,0);
			add((j-1)*n+v,j*n+u,0);
		}
	}
	dijkstra(s);
	cout<<dis[t+k*n]<<endl;
	return 0;
}
2023/5/2 22:29
加载中...