90分WA#11求调and求教
查看原帖
90分WA#11求调and求教
1177472
lcgjyyt楼主2024/10/21 13:39
#include<bits/stdc++.h>
using namespace std;
const int N=5e6+10;
int cnt = 1,dis[N],head[N],vis[N];
struct node{
	int v,nex,w;
}a[N];

void add(int u,int v,int w){
	a[cnt].w=w;
	a[cnt].v=v;
	a[cnt].nex=head[u];
	head[u]=cnt++;
}

void dijkstra(int n,int s){
	priority_queue< pair<int,int> , vector< pair<int,int> > , greater< pair<int,int> > >q;
	memset(dis,0x3f,sizeof(dis));
	q.push(make_pair(0,s));
	dis[s]=0;
	while(!q.empty()){
		pair<int,int> temp=q.top();
		q.pop();
		int u=temp.second;
		if(vis[u]) continue;
		vis[u]=1;
		for(int i=head[u];i!=0;i=a[i].nex){
			int v=a[i].v;
			if(dis[v]>dis[u]+a[i].w){
				dis[v]=dis[u]+a[i].w;
				q.push(make_pair(dis[v],v));
			}
		}
	}
}

int main(){
	int n,m,u,v,w,s,t,k;
	cin>>n>>m>>k>>s>>t;
	memset(head,0,sizeof(head));
	for(int i=0;i<m;i++){
		cin>>u>>v>>w;
		for(int j=0;j<=k;j++){
			add(u+j*n,v+j*n,w);
			add(v+j*n,u+j*n,w);
			if(j!=k){
				add(u+j*n,v+(j+1)*n,0);
				add(v+j*n,u+(j+1)*n,0);
			}
		}
		
	}
	dijkstra(n,s);
	cout<<dis[t+k*n];
	return 0;
}

中间为啥要让上一层的终点连到下一层的起点
(add(v+j*n,u+(j+1)*n,0);)

2024/10/21 13:39
加载中...