POJ 2411 状态压缩DP

原文:http://www.2cto.com/kf/201208/146894.html

编程之美的课后题也有一个和整个题目一样的。(P269)

题目

这个题目的题意很容易理解,在一个N*M的格子里,我们现在有两种类型的 砖块,1 * 2 和 2 * 1,问一共有多少种方案,可以将整个N*M的空间都填满。

最简单的例子就是下面的了:


编程之美中题目:

某年夏天,位于希格玛大厦四层的微软亚洲研究院对办公楼的天井进行了一次大 规模的装修.原来的地板铺有 N×M 块正方形瓷砖,这些瓷砖都已经破损老化了,需要予以 更新.装修工人们在前往商店选购新的瓷砖时,发现商店目前只供应长方形的瓷砖,现在的 一块长方形瓷砖相当于原来的两块正方形瓷砖, 工人们拿不定主意该买多少了, 读者朋友们 请帮忙分析一下:能否用 1×2 的瓷砖去覆盖 N×M 的地板呢?

我们在这里指分析第一个陈述,其实第二个陈述是一样的思路:


这个题目类属于状态压缩DP,对于状态压缩DP,其实最简单的理解就是把状态用比特位的形式表示出来,我们会在下面用例子来说明。

假如现在我们在铺砖 位置(i, j), 并且假设之前的位置已经铺设好的了,在这个位置,我们的选择:

1. 不用铺砖了,可能在(i-1, j)的时刻已经被竖着铺上了,然后考虑的是(i, j+1)
2. 横铺砖,将(i, j+1)也铺上了,然后考虑的是(i, j+2)

3. 竖着铺砖,(将i,j)和(i+1,j)铺上一个竖立的转头。


所以我们如下翻译我们的选择,在位置(i, j) 如果我们选择横着贴砖,那么将(i, j), (i, j+1)都填写成1, 如果竖着贴砖,我们将(i,j)填写成0, 将(i+1, j)填写成1.

为什么要这么计数呢,我觉得应该这样理解:

1. 在横着贴砖的时候,(i, j), (i, j+1) 都是1,这个值其实对下一行如何选择没有影响。

2. 竖着贴砖的第二个,我们也选择了1, 因为这个砖头结束了,对下一行如何选择依然没有影响。

3. 而竖着的第一个砖头,这个砖头是对下面有影响的,如果(i,j)是0,那么(i+1, j)只有是1的情况下才能满足条件。


(这涉及到接下来的 状态兼容性问题)


对于竖着贴砖为什么这样选择,这样选择的一个好处是,我们在处理最后一行的时候,可以保证最后一行都是1, 因为最后一行绝对不能成为 竖砖开始,所以很容易取得最后的解。


好了,我们把这样理解的方案画成图:

 

如果我们将每一行都理解成一个二进制数字,那么

Row1 = 51,  Row2 = 15, Row3 = 48, Row4 = 63, Row5 = 51, Row6 = 63.

最后转头铺满的状态,一定是最后一行全是1。

我们用DP(i,j) 表示如下含义: 当第i行,达到状态j的时候,所能采取的方案数目。 所以明显我们的最后目的是求 DP(N, 2^(M-1)-1);


我们再来简单的分析一下为什么问题可以满足动态规划, 加入现在分析的对象是 DP(i,j), 那么这一行有多少种铺设办法是和上一行相关的,

如果上一行的某个状态DP(i-1,k) 可以达到 DP(i, j) 我们认为这两个状态是兼容的,如果DP(i-1,k)和DP(i, j)兼容并且 DP(i-1, k)有S中铺设方案,那么DP(i, j)就可以从DP(i-1, k)

这条路径中获得S个方案。 当然这里k的取值可以是 0 ~~~~ 2^(M-1) -1种取值。


现在我们来理解一下,什么叫做 j, k 兼容。

其实我们在上面已经基本给出分析, 如果我们现在铺设 (i,x) x这里表示第i行,第x列

1. 如果值 i  行,j 在x位上的值是0, 那么第 i-1行,j的值在x位上一定是1。因为不可能在同一列相邻的位置铺两个竖着的 第一个,如果满足下一步测试的是(i, x+1), 否则直接返回不兼容。


2. 如果值 i  行,j在x位置的值是1 .

{


            那么有可能有两种情况:

            1. (i-1, x)是0, 这个时候一定是竖着铺设了,下一步检测的是(i, x + 1)

           

            2.  (i-1, x) 是1, 如果是这样的话,那么(i, x)一定是要选择横着铺了,那么(i,x+1)也一定是1,并且(i-1, x + 1)一定是1(如果是0,就是竖着铺了),如果不满足就返回不兼容,满足条件 就测试(i, x + 2)

              

}


对于第一行的兼容性,我们要做一下特别的分析,在第一行中,要么放0, 要么放1。

加入当前测试的是 DP(0, j)的第 x的比特位,即第0行,x列

1. 如果x是1,那么 x + 1 也一定是1,然后测试到 x + 2

2. 如果x是0, 那么直接测试下一个 x + 1


补充说明一点,当测试循环中,我们有时候必须要移动 1 位,有时候移动2位,当需要移动2位并且 x == M - 1(M列数)的时候,说明已经不可能兼容了。


根据上面的分析就不难写出代码了:

  1 #include <stdio.h> 
  2 #include <memory.h> 
  3 #include <math.h> 
  4 #include <algorithm> 
  5 using namespace std; 
  6  
  7  
  8 #define MAX_ROW 11 
  9 #define MAX_STATUS 2048 
 10 long long DP[MAX_ROW][MAX_STATUS]; 
 11 int g_Width, g_Height; 
 12  
 13 bool TestFirstLine(int nStatus)  //test the first line 
 14 { 
 15     int i = 0;  
 16     while( i < g_Width) 
 17     { 
 18         if(nStatus & (0x1 << i)) 
 19         { 
 20             if( i == g_Width -1 || (nStatus & (0x1 << (i+1))) == 0) 
 21             { 
 22                 return false; 
 23             } 
 24             i += 2; 
 25         } 
 26         else 
 27         { 
 28             i++; 
 29         } 
 30     } 
 31     return true; 
 32 } 
 33  
 34 bool CompatablityTest(int nStatusA, int nStatusB) // test if status (i, nStatusA) and (i-1, nStatusB) is compatable. 
 35 { 
 36     int i = 0; 
 37  
 38     while( i < g_Width) 
 39     { 
 40         if( (nStatusA & (0x1 << i))  == 0) 
 41         { 
 42             if((nStatusB & (0x1 << i)) == 0) 
 43             { 
 44                 return false; 
 45             } 
 46             i++; 
 47         } 
 48         else 
 49         { 
 50             if((nStatusB & (0x1 << i)) == 0 ) 
 51             { 
 52                 i++; 
 53             } 
 54             else if( (i == g_Width - 1) || ! ( (nStatusA & (0x1 << (i+1))) && (nStatusB & (0x1 << (i + 1)))) ) 
 55             { 
 56                 return false; 
 57             } 
 58             else 
 59             { 
 60                 i += 2; 
 61             } 
 62  
 63         } 
 64     } 
 65     return true; 
 66 } 
 67 int main() 
 68 { 
 69     int i,j; 
 70     int k; 
 71     while(scanf("%d%d", &g_Height, &g_Width) != EOF ) 
 72     { 
 73         if(g_Width == 0 && g_Height == 0) 
 74         { 
 75             break; 
 76         } 
 77  
 78         if(g_Width > g_Height) 
 79         { 
 80             swap(g_Width, g_Height); 
 81         } 
 82  
 83  
 84         int nAllStatus = 2 << (g_Width-1); 
 85         memset(DP, 0, sizeof(DP)); 
 86         for( j = 0; j < nAllStatus; j++) 
 87         { 
 88             if(TestFirstLine(j)) 
 89             { 
 90                 DP[0][j] = 1; 
 91             } 
 92         } 
 93  
 94         for( i = 1; i < g_Height; i++) 
 95         {     
 96             for( j = 0; j < nAllStatus; j++)// iterate all status for line i 
 97             { 
 98                 for( k = 0; k < nAllStatus; k++) // iterate all status for line i-1 
 99                 { 
100                     if(CompatablityTest(j, k)) 
101                     { 
102                         DP[i][j] += DP[i-1][k]; 
103                     } 
104                 } 
105             } 
106         } 
107         printf("%lld
", DP[g_Height-1][nAllStatus - 1]); 
108     } 
109     return 0; 
110 } 
View Code

分析:首先能想到的是如果h*w为奇数,那么肯定不能放满,结果为0

           如何表示状态:从题目看数据很小,那么很容易想到用二进制来表示状态,0表示当前方格没放,1表示放了

           这里用dp[ r ][ cur ]表示前 r-1 行放满,第 r 行状态为 cur 时的方案数,同理,dp[ r-1 ][ lst ]表示前 r-2 行放满,

           第 r-1 行状态为 lst 时的方案数; 状态 lst 是由 cur 状态推出来的( [1] 为什么?),

           所以dp[ r ][ cur ] = dp[ r ][ cur ] +dp[ r-1 ][ lst ]; 那么dp[ h ][ (1<<w)-1 ] 即为最后所求结果

           [1] 先表示一下状态,我们在考虑第 r 行,只考虑前 r 行,一共有 3 种状态(cur表示第 r 行,lst表示第 r-1 行):

                     1、竖直放    cur<<1 | 1 ,   lst<<1 

                     2、横着放    cur<<2 | 3,    lst<<2 | 3

                     3、不放        cur<<1,         lst<<1 | 1

           先解释一下:假设 cur = lst = 0,   假如竖直放,那么 cur = 1, lst = 0, 意思是第 r 行第1列放,第 r-1 行第1列

           不放(因为第 r 行要坚直放,所以第 r-1 行第 1 列不能放),直到放完最后一列,出现了两个状态 cur 和 lst,

           其中 lst 不就是由 cur 状态推出来的么?并且保证第 r-1 行是满的!

           其中第一行只有两种状态,因为第一行上边没有,所以不能竖直放……

           举个例子:h = 3, w = 2;      dp[ r ][ cur ] += dp[ r-1 ][ lst ];

 1 <span style="font-family:Times New Roman;font-size:14px;">#include<iostream>
 2 #include<algorithm>
 3 #include<cstring>
 4 #include<ctime>
 5 #include<cmath>
 6 #include<string>
 7 #include<cstdio>
 8 #include<vector>
 9 using namespace std;
10 
11 int h, w;
12 long long dp[12][2100];
13 
14 void dfs(int r, int c, int cur, int lst) {
15     if(r == 1) { // if r == 1 only have two state
16         if(c == w) { 
17             dp[r][cur]++; 
18             return ; 
19         }
20         if(c + 1 <= w) dfs(r, c+1, cur<<1, lst);
21         if(c + 2 <= w) dfs(r, c+2, cur<<2|3, lst);
22     }
23     else {
24         if(c == w) { // one state occurs
25             dp[r][cur] += dp[r-1][lst];
26             return;
27         }
28         if(c + 1 <= w) { 
29             dfs(r, c+1, cur<<1|1, lst<<1);
30             dfs(r, c+1, cur<<1, lst<<1|1);
31         }
32         if(c + 2 <= w) {
33             dfs(r, c+2, cur<<2|3, lst<<2|3);
34         }
35     }
36 }
37 
38 int main() {
39     while(~scanf("%d%d", &h, &w)) {
40         if(h + w == 0) break;
41         if((h*w) & 1) {
42             puts("0"); continue;
43         }
44         if(h < w) swap(h, w);
45         memset(dp, 0, sizeof(dp));
46         for(int r = 1; r <= h; ++r)
47             dfs(r, 0, 0, 0);
48             
49         printf("%lld
", dp[h][(1<<w)-1]);
50     }
51     return 0;
52 }
53 </span>
View Code
原文地址:https://www.cnblogs.com/qiu520/p/3655913.html