洛谷 P5764 [CQOI2005] 新年好

#include<bits/stdc++.h>
#define int long long
#define PII pair<int,int>
#define se second
#define fi first
#define pb push_back
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
vector<PII>v[200000];
int dist[6][200000];
int vis[200000];
map<int,int>mp;
int n,m,a,b,c,d,e;
int va[200];
void distra(int idx,int A)
{
	priority_queue<PII,vector<PII>,greater<PII>>q;
	for(int i=1;i<=n;i++)
	{
		dist[idx][i]=1e18;
		vis[i]=0;
	}
	q.push({0,A});
	dist[idx][A]=0;
	while(q.size())
	{
		auto t=q.top();
		q.pop();
		int now=t.se;
		if(vis[now]==1)continue;
		vis[now]=1;
		for(auto tt:v[now])
		{
			int spot=tt.se,w=tt.fi;
			if(dist[idx][spot]>dist[idx][now]+w)
			{
				dist[idx][spot]=dist[idx][now]+w;
				q.push({dist[idx][spot],spot});
			}
		}
	}
}
signed main()
{
	IOS;
	cin>>n>>m;
	cin>>a>>b>>c>>d>>e;
	va[0]=1,va[1]=a,va[2]=b,va[3]=c,va[4]=d,va[5]=e;//将亲戚存入数组
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		cin>>x>>y>>z;
		v[x].pb({z,y});
		v[y].pb({z,x});//建立双向边
	}
	sort(va+1,va+1+5);//进行排序，为全排列做准备
	for(int i=0;i<=5;i++)
	{
		distra(i,va[i]);//查找从va[i]（起点）到第i个亲戚的最短路，存入dist[i][va[i]]
		mp[va[i]]=i;
	}
	int minn=1e18;
	do{
		int sum=0;
		for(int i=0;i<=4;i++)
		{
			sum+=dist[mp[va[i]]][va[i+1]];//用map可以在全排列中保持i固定不变
		}
		minn=min(minn,sum);
	}while(next_permutation(va+1,va+1+5));//全排列
    cout<<minn;//输出答案
	return 0;
}