分层图+dij,46pts求调
查看原帖
分层图+dij,46pts求调
879242
SJZ08楼主2024/5/5 12:47

代码:

#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int N=600005,M=6000005;
int h[N],e[M],w[M],ne[M],d[N],idx,n,m,k,v[N];
typedef pair<int,int> PII;
priority_queue<PII,vector<PII>,greater<PII> >que;
void add(int x,int y,int z) {
	e[idx]=y,w[idx]=z,ne[idx]=h[x],h[x]=idx++;
}
void Dijkstra(int sx) {
	memset(d,0x7f,sizeof d);
	d[sx]=0;
	que.push({0,sx});
	while (!que.empty()) {
		PII t=que.top();
		que.pop();
		int x=t.second;
		if (v[x]==1) continue;
		v[x]=1;
		for (int i=h[x];i!=-1;i=ne[i]) {
			int j=e[i];
			if (d[j]>d[x]+w[i]) {
				d[j]=d[x]+w[i];
				que.push({d[j],j});
			}
		}
	}
}
int main() {
	memset(h,-1,sizeof h);
	scanf("%d %d %d",&n,&m,&k);
	int sx,ex;
	scanf("%d %d",&sx,&ex);
	sx++,ex++;
	while (m--) {
		int x,y,z;
		scanf("%d %d %d",&x,&y,&z);
		x++,y++;
		for (int i=0;i<=k;i++) {
			add(x+i*n,y+i*n,z);
			add(y+i*n,x+i*n,z);
			if (i==k) break;
			add(x+i*n,y+(i+1)*n,0);
			add(y+i*n,x+(i+1)*n,0);
		}
	}
	Dijkstra(sx);
	int ans=0x7f7f7f7f;
	for (int i=0;i<=k;i++) {
		ans=min(ans,d[(i+1)*ex]);
	}
	printf("%d",ans);
	return 0;
}
2024/5/5 12:47
加载中...