2个TLE,一个WA,求大佬指点
查看原帖
2个TLE,一个WA,求大佬指点
1088782
Lycdeer楼主2024/8/10 23:23
#include<bits/stdc++.h>
using namespace std;
const long long N=1000000,inf=2147483647;
int n,m,k,s,t,cnt; 
long long to[N],val[N],h[N],nxt[N],dis[N];
bool vis[N];
void add(int a,int b,int c){
	to[++cnt]=b;
	val[cnt]=c;
	nxt[cnt]=h[a];
	h[a]=cnt;
}
struct node{
	int v,w;
	friend bool operator<(node a,node b){
		return a.w>b.w;
	}
}tmp;
priority_queue<node> q;
void Dijkstra(){
	for(int i=0;i<=(k+1)*n;i++) dis[i]=inf;
	dis[s]=0;
	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]=1;
		for(int i=h[u];i;i=nxt[i]){
			if(dis[to[i]]>dis[u]+val[i]){
				dis[to[i]]=dis[u]+val[i];
				tmp.v=to[i]; tmp.w=dis[to[i]]; q.push(tmp); 
			} 
		}
	}
}
int main(){
	scanf("%d%d%d%d%d",&n,&m,&k,&s,&t);
	int a,b,c;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&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*n-n,b+j*n,0);
			add(b+j*n-n,a+j*n,0);
		}
	}
	for(int i=1;i<=k;i++) add(t+(i-1)*n,t+i*n,0); 
	Dijkstra();
	long long ans=inf;
	for(int j=0;j<=k;j++) ans=min(ans,dis[t+j*n]);
	cout<<ans;
	return 0;
}
2024/8/10 23:23
加载中...