旅行商(货担郎)问题求解并求路径的代码

前言

CSDN平台上大部分旅行商的代码要么是有问题的，要么就是教科书PPT的伪代码。正好写了一个，供读者参考（本人水平有限）。

一些提示

1. dfs1用来求哈密顿回路的距离最小和。
2. dfs2根据dfs1求得的res来求具体路径。
3. 两个记忆化搜索都用到了二进制状态压缩的技巧（在状压dp中很常见，更多的位运算技巧可见这里）。
4. 时间复杂度：$O(n^2\cdot 2^n)$

代码

#include <bits/stdc++.h>
using namespace std;

static int N = 4;
// 邻接矩阵
static int w[4][4] = { 
        {0, 6, 7, 9},
        {8, 0, 9, 7},
        {5, 8, 0, 8},
        {6, 5, 5, 0}
};
static int dp[4][1 << 4], DP[4][1 << 4];
static vector<vector<int>> graph(N);	// 邻接表

int dfs1(int u, int i, int mask) {
    if (i == N) return 0;
    if (dp[u][mask] ^ -1) // 记忆化 trick
    	return dp[u][mask];
    int res = INT_MAX;
    for (int v : graph[u])
        if (not (mask >> v & 1))	// v 已经到访过了
            res = min(dfs1(v, i + 1, mask | (1 << v)) + w[u][v], res);
    return dp[u][mask] = res;
}

static int res;
static vector<int> path, PATH;
void dfs2(int u, int i, int mask, int tot) {
    if (i == N) {
        if (not PATH.empty()) return;	// 已经找到路径了
        if (tot == res) PATH = path;
        return;
    }
    if (DP[u][mask]) return;	// 还是记忆化 避免重复计算
    for (int v : graph[u])
        if (not (mask >> v & 1)) {
            path.push_back(v);
            dfs2(v, i + 1, mask | (1 << v), tot + w[u][v]);
            path.pop_back();
        }
    DP[u][mask]++;
}

int main()
{
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            if (w[i][j])
                graph[i].push_back(j);
        }
    }
    memset(dp, -1, sizeof dp);
    res = dfs1(0, 0, 0);

    cout << "距离和：" << res << '\n';
    path.push_back(0); // 起点
    dfs2(0, 0, 0, 0);
    for (int x : PATH) {
        cout << x << ' ';
    }

    return 0;
}