最后一个点WA了
查看原帖
最后一个点WA了
727348
Dreamcatching楼主2023/6/29 16:19
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define int long long
using namespace std;
inline int read(){
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
	return s*w;
}
int sum,n,m,a[10],u,v,w,h[50050],cnt,dis[7][50050],vis[50050],ans=0x3ffff,viss[10];
struct node{
	int to,w,ne;
}E[200005];
void add(int u,int v,int w){
	E[++cnt].to=v;
	E[cnt].w=w;
	E[cnt].ne=h[u];
	h[u]=cnt;
}
void dij(int root){
	memset(dis[root],127-1,sizeof(dis[root]));
	memset(vis,0,sizeof(vis));
	priority_queue<pair<int,int> >q;
	dis[root][a[root]]=0;
	q.push(make_pair(0,a[root]));
	int now;
	while(!q.empty()){
		int now=q.top().second;q.pop();
		if(vis[now]) continue;
		vis[now]=1;
		for(int i=h[now];i;i=E[i].ne){
			int to=E[i].to;
			if(dis[root][to]>dis[root][now]+E[i].w){
				dis[root][to]=dis[root][now]+E[i].w;
				if(!vis[to]) q.push(make_pair(-dis[root][to],to));
			}
		}
	}
	return;
}
void dfs(int use,int now){
	if(use>5){
		ans=min(ans,sum);
		return;
	}
	for(int i=2;i<=6;i++){
		if(viss[i]) continue;
		viss[i]=1;
		sum+=dis[now][a[i]];
		dfs(use+1,i);
		sum-=dis[now][a[i]];
		viss[i]=0;
	}
	return;
}
signed main(){
	n=read(),m=read();
	for(int i=2;i<=6;i++) a[i]=read();
	a[1]=1;
	memset(h,-1,sizeof(h));
	for(int i=1;i<=m;i++){
		u=read(),v=read(),w=read();
		add(u,v,w);
		add(v,u,w);
	}
	for(int i=1;i<=6;i++){
		dij(i);
	}
	dfs(1,1);
	printf("%d",ans);
	return 0;
}
2023/6/29 16:19
加载中...