其他分享
首页 > 其他分享> > 【ACWing】1135. 新年好

【ACWing】1135. 新年好

作者:互联网

题目地址:

https://www.acwing.com/problem/content/1137/

重庆城里有 n n n个车站, m m m条双向公路连接其中的某些车站。每两个车站最多用一条公路连接,从任何一个车站出发都可以经过一条或者多条公路到达其他车站,但不同的路径需要花费的时间可能不同。在一条路径上花费的时间等于路径上所有公路需要的时间之和。佳佳的家在车站 1 1 1,他有五个亲戚,分别住在车站 a , b , c , d , e a,b,c,d,e a,b,c,d,e。过年了,他需要从自己的家出发,拜访每个亲戚(顺序任意),给他们送去节日的祝福。怎样走,才需要最少的时间?

输入格式:
第一行:包含两个整数 n , m n,m n,m,分别表示车站数目和公路数目。第二行:包含五个整数 a , b , c , d , e a,b,c,d,e a,b,c,d,e,分别表示五个亲戚所在车站编号。以下 m m m行,每行三个整数 x , y , t x,y,t x,y,t,表示公路连接的两个车站编号和时间。

输出格式:
输出仅一行,包含一个整数 T T T,表示最少的总时间。

数据范围:
1 ≤ n ≤ 50000 1≤n≤50000 1≤n≤50000
1 ≤ m ≤ 1 0 5 1≤m≤10^5 1≤m≤105
1 < a , b , c , d , e ≤ n 1<a,b,c,d,e≤n 1<a,b,c,d,e≤n
1 ≤ x , y ≤ n 1≤x,y≤n 1≤x,y≤n
1 ≤ t ≤ 100 1≤t≤100 1≤t≤100

思路是Dijkstra + DFS。可以先预处理一下 1 1 1号点以及 5 5 5个亲戚家,两两之间的最短路长度,这可以用Dijkstra算法来做。由于点数和边数数量级差不多,所以是稀疏图,要用堆优化版的Dijkstra来做。预处理完成之后,从 1 1 1号点出发,枚举所有的走遍五个亲戚的路线(一共 5 ! = 120 5!=120 5!=120种排列顺序),找到最小的那个路径长度即可。代码如下:

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

typedef pair<int, int> PII;

const int N = 50010, M = 200010, INF = 0x3f3f3f3f;
int n, m;
// source存6个点的车站编号,source[0]存1,别的分别存a, b, c, d, e
int source[6];
// 邻接表建图的数组
int h[N], e[M], ne[M], w[M], idx;
// dist[i][j]表示source[i]与j的最短距离是多少
int dist[6][N];
bool st[N];
int res = INF;

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

void dijkstra(int start, int dis[]) {
    memset(st, 0, sizeof st);
    priority_queue<PII, vector<PII>, greater<PII> > heap;
    heap.push({0, start});
    dis[start] = 0;

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

        int v = t.second, d = t.first;
        if (st[v]) continue;

        st[v] = true;
        for (int i = h[v]; ~i; i = ne[i]) {
            int j = e[i];
            if (!st[j] && dis[j] > d + w[i]) {
                dis[j] = d + w[i];
                heap.push({dis[j], j});
            }
        }
    }
}

// u表示当前已经走过了多少个点,start表示当前位于的点编号,distance表示已经走了多少距离
void dfs(int u, int start, int distance) {
	// 当走完6个点了,则得到一条路径,更新答案
    if (u == 6) res = min(res, distance);

	// 枚举下一个点走谁
    for (int i = 1; i <= 5; i++)
        if (!st[i]) {
            int next = source[i];
            st[i] = true;
			dfs(u + 1, i, distance + dist[start][next]);
            st[i] = false;
        }
}

int main() {
    cin >> n >> m;
    
    source[0] = 1;
    for (int i = 1; i <= 5; i++) cin >> source[i];

	// 邻接表建图
    memset(h, -1, sizeof h);
    for (int i = 0; i < m; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c), add(b, a, c);
    }

	// 算一下6个点彼此之间的最短路距离
    memset(dist, 0x3f, sizeof dist);
    for (int i = 0; i < 6; i++) dijkstra(source[i], dist[i]);

    memset(st, 0, sizeof st);
    // 从source[0]出发,枚举所有可能的路径
    dfs(1, 0, 0);
    printf("%d\n", res);

    return 0;
}

时间复杂度 O ( m log ⁡ n ) O(m\log n) O(mlogn),空间 O ( n + m ) O(n+m) O(n+m)。

标签:dist,start,int,车站,1135,st,source,新年好,ACWing
来源: https://blog.csdn.net/qq_46105170/article/details/115611235