新年好 —— 最短路 + DFS

题面

传送门

思路

预处理以自己家和$5$个亲戚家为起点的，到其他点的最短路(跑6次迪杰斯特拉)；

接着dfs拜访顺序，一共是$5!$次，并不是很大；

也就是dfs枚举全排列；

Code

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <algorithm>
#include <queue>
#include <utility>
#include <vector>

using namespace std;

typedef long long ll;

const int N = 50000 + 10,M = 1e5+10;

int n,m,tot;

int level[N],w[N];//每个点的等级、直接购买每个点的价格

int head[N],dist[7][N];

bool vis[N];

struct Edge{
    int next,to,val;
}e[M<<1];

void add(int u,int v,int w){
    e[++tot].to = v;
    e[tot].val = w;
    e[tot].next = head[u];
    head[u] = tot;
}

int loc[7];

typedef pair<int,int> pii;

void dij(int start,int dist[]){
    priority_queue<pii,vector<pii>,greater<pii>> pq;
    //clear
    for(int i=1;i<=n;++i) dist[i] = 1e9,vis[i] = 0;
    pq.push({0,start});
    dist[start] = 0;
    while(!pq.empty()){
        int u = pq.top().second;
        pq.pop();
        vis[u] = 1;
        for(int i=head[u];i;i=e[i].next){
            int to = e[i].to;
            if(dist[to] > dist[u] + e[i].val){
                dist[to] = dist[u] + e[i].val;
                pq.push({dist[to],to});
            }
        }
    }
}

int ans;

//全排列
//当前枚举第几个点了，当前从哪个点出发，花费总时间是多少
void dfs(int cnt,int start,int sum){
    if(cnt > 5){
        ans = min(ans,sum);
        return;
    }
    for(int i=2;i<=6;++i){
        if(!vis[i]){
            vis[i] = 1;
            dfs(cnt+1,i,sum+dist[start][loc[i]]);
            vis[i] = 0;
        }
    }
}

void solve(){
    cin >> n >> m;
    loc[1] = 1;
    for(int i=2;i<=6;++i) cin >> loc[i];
    for(int i=1,u,v,w;i<=m;++i){
        cin >> u >> v >> w;
        add(u,v,w);
        add(v,u,w);
    }
    for(int i=1;i<=6;++i) dij(loc[i],dist[i]);
    //枚举拜访顺序
    //即枚举 5！
    memset(vis,0,sizeof vis);
    ans = 1e9;
    dfs(1,1,0);
    cout << ans << '\n';
}

int main(){
    std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
    return 0;
}