旅行商问题(TSP)的 C++ 动态规划解法教学攻略

一、问题描述

旅行商问题(TSP)是一个经典的组合优化问题。给定一个无向图,图中的顶点表示城市,边表示两个城市之间的路径,边的权重表示路径的距离。一个售货员需要从驻地出发,经过所有城市后回到驻地,要求总的路程最短。

二、输入输出形式

输入形式

输入的第一行包含两个整数 nm,分别表示顶点个数和边数。接下来的 m 行中,每行包含三个整数 uvw,表示顶点 u 和顶点 v 之间有一条边,边的权重为 w

输出形式

输出一个整数,表示最短路程的总长度。

三、样例输入输出

样例输入

复制代码
5 10
1 2 3
1 3 1
1 4 5
1 5 8
2 3 6
2 4 7
2 5 9
3 4 4
3 5 2
4 5 3

样例输出

16

四、算法分析与设计

1. 动态规划算法

旅行商问题可以通过动态规划(DP)来解决。其核心思想是利用状态压缩来记录已经访问过的城市集合,并使用动态规划表来存储到达每个城市的最短路径。

2. 状态表示

定义 dp[S][i] 表示已经访问了集合 S 中的城市,最后到达城市 i 的最短路径。其中,S 是一个位掩码,表示已经访问过的城市集合。例如,S = (1 << n) - 1 表示所有城市都被访问过。

3. 状态转移

对于每个状态 S,遍历所有可能的城市 i,如果 i 在集合 S 中,则尝试将未访问的城市 j 加入集合 S,更新新的状态 S | (1 << j) 的最短路径。

4. 初始状态和结果提取

初始状态为 dp[full][i] = dist[i][0],其中 full 表示所有城市都被访问过,dist[i][0] 是城市 i 回到起点 0 的距离。最终结果从 dp[1][0] 中提取,表示从起点 0 出发,访问所有城市后回到起点的最短路径。

五、代码实现

复制代码
#include <iostream>
#include <vector>
#include <algorithm>
#include <climits>
using namespace std;

const int INF = 0x3f3f3f3f;

int main() {
    int n, m;
    cin >> n >> m;

    // 初始化邻接矩阵
    vector<vector<int>> dist(n, vector<int>(n, INF));
    for (int i = 0; i < n; i++) {
        dist[i][i] = 0;
    }

    // 输入边信息并构建邻接矩阵
    for (int i = 0; i < m; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        u--; // 转换为从 0 开始的索引
        v--;
        dist[u][v] = w;
        dist[v][u] = w;
    }

    // 动态规划表
    int full = (1 << n) - 1;
    vector<vector<int>> dp(1 << n, vector<int>(n, INF));

    // 初始化动态规划表,full 表示所有城市都被访问过
    for (int i = 0; i < n; i++) {
        dp[full][i] = dist[i][0];
    }

    // 填充动态规划表
    for (int S = full; S >= 0; S--) {
        if ((S & 1) == 0) continue; // 必须包含起点(城市 0)

        for (int i = 0; i < n; i++) {
            if (!((S >> i) & 1)) continue; // 城市 i 不在集合 S 中时跳过

            if (S == full) {
                continue; // full 已经初始化
            }

            for (int j = 0; j < n; j++) {
                if ((S >> j) & 1) continue; // 城市 j 已经在集合 S 中时跳过

                dp[S][i] = min(dp[S][i], dp[S | (1 << j)][j] + dist[i][j]);
            }
        }
    }

    // 输出结果
    cout << dp[1][0] << endl;

    return 0;
}

六、代码解析

1. 邻接矩阵初始化

首先,我们创建一个大小为 n x n 的邻接矩阵 dist,初始化所有元素为 INF(表示无穷大)。然后将对角线元素设置为 0,因为每个城市到自身的距离为 0。接下来,读取边信息并更新邻接矩阵。

2. 动态规划表初始化

定义动态规划表 dp,其中 dp[S][i] 表示已经访问了集合 S 中的城市,最后到达城市 i 的最短路径。初始状态为 dp[full][i] = dist[i][0],其中 full 表示所有城市都被访问过,此时需要回到起点 0

3. 状态转移填充

full 开始,逐步减少集合的大小,填充动态规划表。对于每个状态 S 和城市 i,如果 i 在集合 S 中,则尝试将未访问的城市 j 加入集合 S,更新新的状态 S | (1 << j) 的最短路径。

4. 输出结果

最终结果从 dp[1][0] 中提取,表示从起点 0 出发,访问所有城市后回到起点的最短路径。

七、复杂度分析

1. 时间复杂度

动态规划的状态数为 O(2^n * n),每个状态需要遍历所有可能的城市 n,因此总时间复杂度为 O(2^n * n^2)

2. 空间复杂度

动态规划表的空间复杂度为 O(2^n * n),用于存储每个状态和城市的最短路径。

八、总结

旅行商问题是一个典型的组合优化问题,通过动态规划算法可以有效地解决。该算法利用状态压缩和动态规划表,能够在合理的时间内找到最短路径。对于较小规模的问题,这种方法是可行的,但对于大规模问题,可能需要更高效的算法或启发式方法。

希望本攻略能够帮助你理解旅行商问题的动态规划解法,并能够在实际问题中应用。如果有任何疑问或需要进一步的解释,请随时提问!

相关推荐
music&movie22 分钟前
算法工程师认知水平要求总结
人工智能·算法
秦少游在淮海27 分钟前
C++ - string 的使用 #auto #范围for #访问及遍历操作 #容量操作 #修改操作 #其他操作 #非成员函数
开发语言·c++·stl·string·范围for·auto·string 的使用
const54435 分钟前
cpp自学 day2(—>运算符)
开发语言·c++
虾球xz1 小时前
CppCon 2015 学习:CLANG/C2 for Windows
开发语言·c++·windows·学习
laocui11 小时前
Σ∆ 数字滤波
人工智能·算法
CodeWithMe1 小时前
【C/C++】namespace + macro混用场景
c语言·开发语言·c++
yzx9910131 小时前
Linux 系统中的算法技巧与性能优化
linux·算法·性能优化
全栈凯哥2 小时前
Java详解LeetCode 热题 100(26):LeetCode 142. 环形链表 II(Linked List Cycle II)详解
java·算法·leetcode·链表
全栈凯哥2 小时前
Java详解LeetCode 热题 100(27):LeetCode 21. 合并两个有序链表(Merge Two Sorted Lists)详解
java·算法·leetcode·链表
SuperCandyXu2 小时前
leetcode2368. 受限条件下可到达节点的数目-medium
数据结构·c++·算法·leetcode