Acwing1135.新年好

重庆城里有 $n$ 个车站，$m$ 条 双向 公路连接其中的某些车站。

每两个车站最多用一条公路连接，从任何一个车站出发都可以经过一条或者多条公路到达其他车站，但不同的路径需要花费的时间可能不同。

在一条路径上花费的时间等于路径上所有公路需要的时间之和。

佳佳的家在车站 $1$，他有五个亲戚，分别住在车站 $a,b,c,d,e$。

过年了，他需要从自己的家出发，拜访每个亲戚（顺序任意），给他们送去节日的祝福。

怎样走，才需要最少的时间？

输入格式

第一行：包含两个整数 $n,m$，分别表示车站数目和公路数目。

第二行：包含五个整数 $a,b,c,d,e$，分别表示五个亲戚所在车站编号。

以下 $m$ 行，每行三个整数 $x,y,t$，表示公路连接的两个车站编号和时间。

输出格式

输出仅一行，包含一个整数 $T$，表示最少的总时间。

数据范围

$1\le n\le 50000,1\le m\le 105,1<a,b,c,d,e\le n,1\le x,y\le n,1\le t\le 100$

输入样例：

6 6
2 3 4 5 6
1 2 8
2 3 3
3 4 4
4 5 5
5 6 2
1 6 7

输出样例：

21

图论加dfs，对于每个车站都做一遍dijkstra，爆搜一遍所有情况即可，sizeof无法确定数组为参数大小，所以需要memet中的大小为N * 4

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#define x first
#define y second

using namespace std;

const int N = 50010, M = 2e5 + 10;
typedef pair<int, int> PII;

int star[6];
int n, m;//车站和公路数目
int h[N], e[M], ne[M], w[M], idx;
int dist[6][N];
bool st[N];
int res = 2e9;

void add(int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

void dfs(int u, int start,int distance)
{
    if(distance >= res) return;
    if(u >= 5) res = min(res, distance);

    for(int i = 1; i <= 5; i ++)
    {
        int next = star[i];
        if(!st[i])
        {
            st[i] = true;
            dfs(u + 1, i, distance + dist[start][next]);
            st[i] = false;
        }
    }
}

void dijkstra(int start, int dist[])
{
    priority_queue<PII, vector<PII>, greater<PII> > heap;

    memset(dist, 0x3f, N * 4);
    memset(st, 0, sizeof st);

    dist[start] = 0;
    heap.push({0, start});

    while(heap.size())
    {
        int t = heap.top().y;
        heap.pop();

        if(st[t]) continue;
        st[t] = true;

        for(int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(dist[j] > dist[t] + w[i])
            {
                dist[j] = dist[t] + w[i];
                heap.push({dist[j], j});
            }
        }
    }   
}

int main()
{
    scanf("%d%d", &n, &m);
    memset(h, -1, sizeof h);
    star[0] = 1;
    for(int i = 1; i <= 5; i ++) scanf("%d", &star[i]);

    while(m --)
    {
        int x, y, t;
        scanf("%d%d%d", &x, &y, &t);
        add(x, y, t), add(y, x, t);
    }

    for(int i = 0; i <= 5; i ++) dijkstra(star[i], dist[i]);

    memset(st, 0, sizeof st);

    dfs(0, 0, 0);

    cout << res << endl;

    return 0;   
}