旅行商问题 状压DP
旅行商问题描述
现在有一个旅行商,在一个国家做生意。这个国家有\(N(2 \le N \le 15)\)个城市,城市之间有单行道可以通行,每条路都有相应的路费\((0 \le d(i,j) \le 1000)\)。现在旅行商要从0号城市出发,经过所有的城市,最后回到0号城市。要求所花的路费最小。保证这些道路能构成一个环(可以一笔画)。
问题简化
给出一张带权有向有环图,用邻接矩阵表示,\(d(i,j)\)表示\(ij\)两个节点之间边的权值,INF表示没有边。要求从0号节点出发,经过每一个节点后正好回到0号节点,问经过边的总权值最小是多少。
问题分析
看到这道题,你的第一反应是不是暴力呀?嗯,很正常,当时我也是这么想的。那我们不妨就来思考一下怎么暴力?其实呀,暴力就是一个全排列问题。把除0号节点外的所有节点全排列一遍,然后排除那些根本走不通的方案,最后在剩下的方案中找最小的花费。
嗯。这种方案不是不行,时间复杂度是\(O((n-1)!)\),在这里n最大是15,那最坏情况下的计算次数是87,178,291,200。超出了int的表示范围。但是这样还是可以骗分的。
进一步分析
其实这个问题是NP难问题。不能在多项式时间内进行求解。但是这种问题是可以用DP的思路来求解的。我们定义一个集合S,表示已经走过的节点的集合,定义集合V表示所有节点。再有v表示当前的节点。定义\(dp[S][v]\) 表示已走过节点为\(S\),当前节点为\(v\)时,从\(v\)出发,经过~S的节点回到0号节点的最小花费。再定义\(u\)表示下一步要走到的节点,\(u\notin S\),可以得到递推式
状态压缩
我们发现,在\(dp\)这个数组中有一个维度是表示状态的集合。但我们又知道,数组的下标必须是整数,那怎么办呢?
我们发现,一个节点要么在,要么不再S集合中。那么n个节点就可以表示成:“在、不在、在、不在S集合中”。再利用计算机二进制存储的特性,我们就可以把这种状态压缩到一个整数当中。而且我们发现\(N \le 15\),也就是说这个表示状态的整数最大不过\(2^{16}-1\),就是65535,保证内存hold住。这种方法叫做状态压缩。(接下来的很多关于位运算的知识,可以参考我的这篇文章:位运算)
状压DP
状压完了,就要开始DP了。DP最重要的条件是什么?最有子结构。这里的子结构一定是最优的(参见01背包的推导方法)。问题是怎么保证子结构呢?假设有两个压缩后的状态\(ij\),当\(S_i\subseteq S_j\)时,\(i\le j\)。这就保证了一定能用循环求解。代码是这样的:
int dp[1 << MAX_N][MAX_N]; //DP数组
void solve() {
//初始化
for (int S = 0; S < 1 << n; ++S)
for (int i = 0; i < n; ++i)
dp[S][i] = INF;
dp[(1 << n) - 1][0] = 0;
//DP
for (int S = (1 << n) - 2; S >= 0; S--) {
for (int v = 0; v < n; ++v) {
for (int u = 0; u < n; ++u) { //遍历u
if (!(S >> u & 1)) //没有访问过
dp[S][v] = min(dp[S][v], dp[S | 1 << u][u] + d[v][u]);
}
}
}
cout << dp[0][0] << endl;
}
现在来解释一下。
我们先假设\(S\)确定,来解释一下里面的循环。\(v\)循环是标准的DP循环,也可以参考01背包的解释。
u循环的意思是遍历当当前节点是v时,下一步的走法。这时候你可能会产生疑问,为什么没有判断v u之间有没有边。因为如果v u之间没有边,那v u之间的距离就是INF,是不会被min****选中的。
min判断的前面表示不选u节点作为下一步,后面表示往后走一步。其实就是找最小值。自己理解一下。
然后我们来理解一下S循环。你可能会产生疑问,为什么S要逆序。01背包逆序是要防止数据的覆盖,但是这本来就是二维数组,不存在这种问题。
要解释这个问题,我们就要先回顾一下动态规划的基本思路,就是有小规模的问题合并成大规模的问题。那什么是小规模的问题?是\(S\in \emptyset\)吗?当然不是,如果\(S\in \emptyset\)的话,就是一个点也没有走,这应该是最后要求解的答案(回顾一下\(dp[S][v]\)的定义)。规模最小的问题应该是已经走过了其他所有的点,就差一步就回到0号节点的状态,就是(1 << n) – 2
。因为已知\(dp[V][0]=0\),所以这种DP是可行的。