动态规划:状态压缩 (蒙德里安的梦想、最短Hamilton路径)

状态压缩DP

AcWing 291.蒙德里安的梦想

求把N* M的棋盘分割成若干个1*2的的长方形,有多少种方案。
例如当N=2,M=4时,共有5种方案。当N=2,M=3时,共有3种方案。
如下图所示:
在这里插入图片描述
输入格式
输入包含多组测试用例。
每组测试用例占一行,包含两个整数N和M。
当输入用例N=0,M=0时,表示输入终止,且该用例无需处理。

输出格式
每个测试用例输出一个结果,每个结果占一行。

数据范围
1≤N,M≤11
输入样例:
1 2
1 3
1 4
2 2
2 3
2 4
2 11
4 11
0 0
输出样例:
1
0
1
2
3
5
144
51205

Code:
f[i][j]表示第i列按状态j摆放时,从i-1列(状态k)转移过来的方案数
第二维的状态中0表示竖着放,1表示横着放
第i-1列状态是否能转移到第i列的条件:
①第i列状态j和第i-1列状态k不能冲突 (j & k) == 0;
②第i列状态j确定后,第i-1列状态k的连续空白格子的长度必须是偶数 st[ j | k] = true.

初始状态:f[0][0] = 1, 其余全为0;第0列只能是全0的状态
最终状态:f[m][0], 棋盘有0~(m-1)列,但后面一列会影响前一列,多出的一列必须是全0.

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 12, M = 1 << N;

int n, m;
long long f[N][M];
bool st[M]; //判断是否存在连续的偶数个0 

int main()
{
    while (cin >> n >> m, n || m)
    {
        for (int i = 0; i < 1 << n; i ++ )
        {
            int cnt = 0;
            st[i] = true;
            for (int j = 0; j < n; j ++ )
                if (i >> j & 1)//取出的数i第j位数字为1
                {
                    if (cnt & 1) //cnt为奇数,不满足转移条件②
                    	st[i] = false;
                    cnt = 0;
                }
                else cnt ++ ;
            if (cnt & 1) st[i] = false;
        }

        memset(f, 0, sizeof f);
        f[0][0] = 1;
        for (int i = 1; i <= m; i ++ ) 
            for (int j = 0; j < 1 << n; j ++ )
                for (int k = 0; k < 1 << n; k ++ )
                    if ((j & k) == 0 && st[j | k])
                        f[i][j] += f[i - 1][k];

        cout << f[m][0] << endl;
    }
    return 0;
}

AcWing 91. 最短Hamilton路径

给定一张 n 个点的带权无向图,点从 0~n-1 标号,求起点 0 到终点 n-1 的最短Hamilton路径。 Hamilton路径的定义是从 0 到 n-1 不重不漏地经过每个点恰好一次。

输入格式
第一行输入整数n。
接下来n行每行n个整数,其中第 i 行第 j 个整数表示点 i 到 j 的距离(记为a[i,j])。
对于任意的x,y,z,数据保证 a[x,x]=0,a[x,y]=a[y,x] 并且 a[x,y]+a[y,z]>=a[x,z]。

输出格式
输出一个整数,表示最短Hamilton路径的长度。

数据范围
1≤n≤20
0≤a[i,j]≤107
输入样例:
5
0 2 4 5 1
2 0 6 5 3
4 6 0 8 3
5 5 8 0 5
1 3 3 5 0
输出样例:
18

Code:
f[i, j]表示从起点0到第 j 点,经过点为 i 的所有路径的距离最小值。
i 表示二进制压缩状态,存储经过哪些节点的信息, j 表示从第0点走到第 j 点。
初始状态为f [0][j] = +∞, f[1][0] = 0; 最终问题解为f[“11…1”, n - 1]即 f[2n-1][n -1].
状态转移:将压缩状态为 i 的路径0~j分为0→k(不包括k)和k→ j。
f[i][j] = min(f[[i - (1<< j)][k] +w[k][j]); ("i - (1 << j)"确保该状态第j位为0,不经过j)

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 20, M = 1 << N;
int n;
int w[N][N];
int f[M][N];

int main()
{
    cin >> n;
    for(int i = 0; i < n; i ++)
        for(int j = 0; j < n; j ++)
            cin >> w[i][j];
    
    memset(f, 0x3f, sizeof f);
    f[1][0] = 0;
    for(int i = 0; i < 1 << n; i ++)
        for(int j = 0; j < n; j ++)
            if(i >> j & 1)
                for(int k = 0; k < n; k ++)
                    if(i >> k & 1)
                        f[i][j] = min(f[i][j], f[i - (1 << j)][k] + w[k][j]);
        
    cout << f[(1 << n) - 1][n - 1] << endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值