3. 更复杂的动态规划

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。

猜你喜欢

转载自www.cnblogs.com/astonc/p/10697269.html
3.