TLE80pts spfa求调
查看原帖
TLE80pts spfa求调
784175
zhujiajun2013楼主2025/2/8 09:57
#include<bits/stdc++.h>
using namespace std;
int n,m,a[6],head[100005],dis[10][100005],cnt;
struct node{
	int to,nxt,w;
}e[200005];
void add(int u,int v,int w){
	e[cnt].to=v;
	e[cnt].w=w;
	e[cnt].nxt=head[u];
	head[u]=cnt++;
}
inline int read(){
    int x=0,k=1; char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')k=-1;c=getchar();}
    while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*k;
}
void spfa(int start){
    queue<int> q;
	q.push(a[start]);
	dis[start][a[start]]=0;
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int i=head[u];~i;i=e[i].nxt){
			int v=e[i].to;
			if(dis[start][v]>dis[start][u]+e[i].w){
				dis[start][v]=dis[start][u]+e[i].w;
				q.push(v);
			}
		}
	}
}
int main(){
	n=read();m=read();
	memset(head,-1,sizeof head);
	a[0]=1;
	for(int i=1;i<=5;i++)a[i]=read();
	for(int i=1;i<=m;i++){
		int u,v,w;
		u=read();v=read();w=read();
		add(u,v,w);add(v,u,w);
	}
	memset(dis,0x3f,sizeof dis);
	bfs(0);
	for(int i=1;i<=5;i++)spfa(i);
	int minn=1e9;
	for(int i1=1;i1<=5;i1++){
		for(int i2=1;i2<=5;i2++){
			if(i1==i2)continue;
			for(int i3=1;i3<=5;i3++){
				if(i1==i3||i2==i3)continue;
				for(int i4=1;i4<=5;i4++){
					if(i1==i4||i2==i4||i3==i4)continue;
					for(int i5=1;i5<=5;i5++){
						if(i1==i5||i2==i5||i3==i5||i4==i5)continue;
						int p=dis[0][a[i1]]+dis[i1][a[i2]]+dis[i2][a[i3]]+dis[i3][a[i4]]+dis[i4][a[i5]];
						minn=min(minn,p);
					}
				}
			}
		}
	}
	cout<<minn;
	return 0;
}
2025/2/8 09:57
加载中...