蒟蒻28pts代码求调
查看原帖
蒟蒻28pts代码求调
1140118
wanghaolin44243787楼主2024/8/8 10:58
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 1000006
struct node{
	ll to,v;
}o;
vector<node>edge[MAXN];
priority_queue<pair<ll,ll>,vector<pair<ll,ll> >,greater<pair<ll,ll> > >pai;
pair<ll,ll>q,p;
ll n,m,k,s,t,x,y,z,dis[MAXN],ans=0x7fffffff;
bool cmp[MAXN];
inline void add(ll a,ll b,ll c){
	o.v=c;
	o.to=b;
	edge[a].push_back(o);
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>k>>s>>t;
	for(int i=1;i<=m;i++){
		cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
		for(int j=1;j<=k;j++){
			add(x+n*(j-1),y+n*j,0);
			add(y+n*(j-1),x+n*j,0);
			add(x+n*j,y+n*j,z);
			add(y+n*j,x+n*j,z);
		}
	}
	for(int i=1;i<=n*(k+1);i++){
		dis[i]=0x7fffffff;
	}
	for(int i=1;i<=k;i++){
		add(t+(i-1)*n,t+i*n,0);
	}
	dis[s]=0;
	q.first=0;
	q.second=s;
	pai.push(q);
	while(pai.empty()==0){
		q=pai.top();
		pai.pop();
		if(cmp[q.second])continue;
		cmp[q.second]=1;
		for(int i=1;i<edge[q.second].size();i++){
			if(dis[edge[q.second][i].to]>dis[q.second]+edge[q.second][i].v){
				dis[edge[q.second][i].to]=dis[q.second]+edge[q.second][i].v;
				p.first=dis[edge[q.second][i].to];
				p.second=edge[q.second][i].to;
				pai.push(p);
			}
		}
	}
	for(int i=0;i<=k;i++){
		ans=min(ans,dis[t+n*i]);
	}
	cout<<ans;
	return 0;
}
2024/8/8 10:58
加载中...