AcWing 1135. 新年好(堆优化Dijkstra + dfs)

传送门

大致题意 : 有n个站点, 和m条双向边, 起点为1, 有5个站点a,b,c,d,e, 问怎么走能是走完5个站点的时间最短

思路 : 暴力做法为, 对起点和站点排序, dfs一遍所有顺序, 对每一种顺序求一遍两点间的最短路, 在求和取最小, 时间复杂度为5! * O(m)(spfa的复杂度), 加上spfa的常数k, 大概率会被卡掉, 换一种思路, 我们可以先预处理一下点对间的最短路, 再进行dfs, 这样的化话时间复杂度会有乘算优化为加算, 即O(m) + 5!, 这样就能过(但是数据加强后会卡spfa, 所以要用堆优化dijkstra过)

代码如下:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <iterator>
#include <cmath>
#include <algorithm>
#include <sstream>
#include <map>
#include <stack>
#include <set>
#include <queue>
#include <iomanip>
using namespace std;

stringstream ss;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 50010, M = 200010;
const int INF = 0x3f3f3f3f;

int n,m;
int sc[6];
int h[N], e[M], ne[M], w[M], idx;
int d[6][N];
bool st[N];

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

void dijkstra(int s, int d[])
{
    // 初始化N*4是因为d[][]为二维数组, 每次初始化只初始化当前这一层(或者在dijkstra外初始化所有的d数组也可)
    memset(d, 0x3f, N*4);
    memset(st, 0, sizeof st);
    d[s] = 0;
    priority_queue<PII, vector<PII>, greater<PII> > heap;
    heap.push({0, s});
    while(heap.size())
    {
        auto t = heap.top();
        heap.pop();
        int ver = t.second;
        if(st[ver]) continue;
        st[ver] = 1;

        for(int i = h[ver]; ~i; i = ne[i])
        {
            int j = e[i];
            if(d[j] > d[ver] + w[i])
            {
                d[j] = d[ver] + w[i];
                heap.push({d[j], j});
            }
        }
    }
}

// dfs寻找最短答案, u为结点数目, s为当前点, d为累积距离
int dfs(int u, int s, int dist)
{
    // 如果节点数目为第6个时, 直接返回累计距离(因为是最后一个点)
    if(u > 5) return dist;

    int res = INF;

    for(int i = 1; i<=5; i++)
    {
        if(!st[i])
        {
            // next为下一个点
            int next = sc[i];
            st[i] = 1;
            res = min(res, dfs(u+1, i, dist+d[s][next]));
            st[i] = 0;
        }
    }
    return res;
}
int main()
{
    scanf("%d%d", &n, &m);
    sc[0] = 1;
    for(int i = 1; i<=5; i++) scanf("%d", &sc[i]);
    memset(h, -1, sizeof h);
    while(m -- )
    {
        int a,b,c;
        scanf("%d%d%d", &a, &b, &c);
        add(a,b,c), add(b,a,c);
    }
//    memset(d, INF, sizeof d);
    for(int i = 0; i < 6; i++) dijkstra(sc[i], d[i]);
    memset(st, 0, sizeof st);
    printf("%d\n", dfs(1, 0, 0));
}