1. 状态压缩DP
这个问题是著名的旅行商问题(TSP,Traveling Salesman Problem)。TSP问题是NP困难的,没有已知的多项式时间的高效算法可以解决这一问题。在这个问题中,所有可能的路线共有(n - 1)!种, 所以肯定不能遍历每一种情况,我们试着用DP来解决。
定义: S : 为现在已经访问过的顶点的集合(起点 0 当做还未访问过的顶点)
v : 为当前所在的顶点
dp[ S ][ v ] =:从 v 出发访问剩余所有的顶点,最终回到顶点 0 的路径的权重总和的最小值。
由于从 v 出发可以移动到任意的一个节点 u ∉ S,递推式为:
dp[ V ][ 0 ] = 0
dp[ S ][ v ] = min { dp[ S υ { u }][ u ] + d( v, u ) | u ∉ S}
在这个递推式中有一个是集合而不是整数,因此需要稍加处理。首先我们使用记忆化搜索求解。虽然有一个是集合, 但是我们可以把它编码为一个整数,或者给它们定义一个全序关系并用二叉搜索树存储。特别地,对于集合我们可以把每一个元素的选取与否对应到一个二进制位里,从而把状态压缩成一个整数,大大方便了计算和维护。
int n; int d[MAX_N][MAX_N]; int dp[1 << MAX_N][MAX_N]; //已经访问过的节点集合为S,当前位置为 v int rec(int S, int v) { if (dp[S][v] >= 0) return dp[S][v]; if (S == (1 << n) - 1 && v == 0) //已经访问过所有节点并回到 0 号点 return dp[S][v] = 0; int res = INF; for (int u = 0; u < n; u++) if (!(S >> u & 1)) res = min(res, rec(S | 1 << u, u) + d[v][u]); return dp[S][v] = res; } void solve() { memset(dp, -1, sizeof(dp)); printf("%d\n", rec(0,0)); }
复杂度为 0(2n n2)。对于不是整数的情况,很多时候很难确定一个合适的递推顺序,因此使用记忆化搜索可以避免这个问题。不过在这个问题中,对于任意两个整数 i 和 j,如果它们对应的集合满足 S(i) ⊆ S(j),就有 i ≤ j,因此可以像下面一样用循环求解。
int n; int d[MAX_N][MAX_N]; int dp[1 << MAX_N][MAX_N]; void solve() { // 用足够大的值初始化数组 for (int S = 0; S < 1 << n; S++) fill(dp[S], dp[S] + n, INF); dp[(1 << n) - 1][0] = 0; for (int S = (1 << n) - 2; S >= 0; S--) for (int v = 0; v < n; v++) for(int u = 0; u < n; u++) if (!(S >> u &1)) dp[S][v] = min(dp[S][v], dp[S | 1 << n][u] + d[v][u]); printf("%d\n", dp[0][0]); }
像这样针对集合的DP , 我们一般叫状态压缩DP。