P5764最后一个节点超时!怎么优化?求助
  • 板块题目总版
  • 楼主Ray_K
  • 当前回复8
  • 已保存回复8
  • 发布时间2024/8/28 14:30
  • 上次更新2024/8/28 16:50:14
查看原帖
P5764最后一个节点超时!怎么优化?求助
705969
Ray_K楼主2024/8/28 14:30
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int h[N],ne[2*N],e[2*N],w[2*N],pt[10]={1},idx;
int n,m,d[10][N];
bool vis[N],st[N];
void adde(int u,int v,int c){
	e[idx]=v;
	w[idx]=c;
	ne[idx]=h[u];
	h[u]=idx++;
}
void spfa(int sx,int d[]){
	queue<int>q;
	memset(vis,false,sizeof vis);
	vis[sx]=true;
	d[sx]=0;
	q.push(sx);
	while(q.size()){
		int u=q.front();
		q.pop();
		vis[u]=false;
		for(int i=h[u];~i;i=ne[i]){
			int v=e[i];
			if(d[v]>d[u]+w[i]){
				d[v]=d[u]+w[i];
				if(!vis[v]) q.push(v),vis[v]=true;
			}
		}
	}
}
int dfs(int u,int start,int dis){
	if(u>5) return dis;
	int res=1e9;
	for(int i=1;i<=5;i++){
		if(!st[i]){
			st[i]=true;
			res=min(res,dfs(u+1,i,dis+d[start][pt[i]]));
			st[i]=false;
		}
	}
	return res;
}
int main(){
	cin>>n>>m;
	memset(h,-1,sizeof h);
	for(int i=1;i<=5;i++) scanf("%d",pt+i);
	for(int i=1;i<=m;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		adde(x,y,z),adde(y,x,z);
	}
	memset(d,0x3f,sizeof d);
	for(int i=0;i<6;i++) spfa(pt[i],d[i]);
	cout<<dfs(1,0,0);
	return 0;
}

代码如上 如果能顺便讲解一下就更好了上课跟着老师写的

2024/8/28 14:30
加载中...