首页 > 代码库 > 【原创】【状态压缩DP】POJ3254 Corn Fields【新手向】

【原创】【状态压缩DP】POJ3254 Corn Fields【新手向】

一开始根本不会状压dp,上网各种找题解,但发现他们写的都很......反正我作为一个没有接触过状态压缩的,根本看不懂!

 

然后看了好多状态压缩的题的题解,总结了一下思路,思路很重要,有了思路转换成计算机语言就好了。因此我先讲一下思路:

  先说说地图,地图上每一行的01代表一个状态,比如输入样例中的111、010,表示第一行的三个位置都可以种稻子,第二行中间的位置可以种稻子,然后,不能种稻子的地方一定不能种稻子(废话...)

可以种稻子的地方可以选择种也可以选择不种,然后有一个前提条件,就是上下左右相邻的地方不能种稻子。

  再说说怎么状态压缩,状态压缩就是把每一个状态压缩成二进制,二进制就是由01组成的,0代表不种,1代表种。二进制就要牵扯到位运算,位运算我就不想说了,百度吧。因此,一串01的二进制数就

可以代表一个状态,例如输入样例第一行是111,那么可以放入第一行的状态有,100、010、001、101、000,因为相邻位置不能放所以只有5种方法,那么第二行就只有2种方法000、010(不考虑其他行)

  那么看第一行和第二行(第一行——第二行),100——000,010——000,001——000,101——000,000——000,这是5种对应方法,还可以100——010,001——010,101——010,000——010这是另外的4种对应方法(第一行5种状态对吧?第二行2种状态,按照乘法原理,应该有5*2 = 10种方法,但是111——010是不合法的,因此样例的答案是10-1 = 9)。

dp[i][j]意思是推到第i行状态为j的方案总数。

那么“100——000”即为dp[2][000]可以由dp[1][100]得到,那么dp[2][000] = dp[2][000] + dp[1][100];

那么“010——000”即为dp[2][000]可以由dp[1][010]得到,那么dp[2][000] = dp[2][000] + dp[1][010];

......

以此类推,逐行递推。

  总结一下思路:先枚举第一行,把所有可能的状态和第一行的地图对比,如果成功,则在循环里继续枚举第二行,把所有可能的状态和第二行的地图对比,如果成功,再和第一行填入的状态对比,如果又匹配成功,则dp[2][000] = dp[2][000] + dp[1][100];方法数加到第二行。这就是一次循环结束了,从新枚举第二行...

把思路转换成代码

can[]代表可行的状态,稍后解释。cur[i]代表地图的第i行
1
for(int i=1;i<m;i++)//枚举每一行 2 { 3 for(int j=0;j<tot;j++)//对第i行枚举所有可行的状态j 4 { 5 if((can[j]&cur[i])==0)//如果状态j和第i行匹配了 6 { 7 for(int k=0;k<tot;k++)//枚举第i+1行的所有可行的状态k 8 { 9 if(((can[k]&cur[i+1])==0)&&((can[k]&can[j])==0))//状态k和第i+1行匹配且和状态j匹配10 dp[i+1][can[k]] = dp[i+1][can[k]]+dp[i][can[j]];//状态数相加11 }12 }13 }14 }

这样核心代码就实现了。

有一个小方法,就是枚举可行状态的时候,假如一行是8列,不必从00000000枚举到11111111,这样很麻烦,所以要预处理。

就是在一开始把,一行的可行状态先求出来就拿“11111111”来说,这肯定是不可能的,因为有相邻的1,所以在一开始就可以舍弃掉。怎么做呢?

假如一行是8列,先从00000000枚举到11111111,对于每一个状态把它左移1位,再和他自己&运算,假如结果>0,就说明有有相邻的1,举个简单的例子:

  01011要判断有没有相邻的1,if(((01011<<1) & (01011)) > 0 )则有相邻的1,(01011<<1) & (01011) 就是 010110和01011按位且运算,这两个红色地方1&1 == 1,因此结果大于0。

怎么实现呢?

1 tot = 0;//全局变量,相当于栈的top,代表可行的状态数2 for(int i=0;i<(1<<n);i++)//n是列数,i是枚举的状态3      if((i&(i<<1))==0) can[tot++] = i;

  dp[][]肯定要初始化对吧?不然全是0了,只要对第一行初始化就行了,因为后面的的行都是由第一行得来的

1 for(int i=0;i<tot;i++)2      if((cur[1]&can[i])==0) dp[1][can[i]] = 1;//和cur[1](第一行)匹配,就给对应的dp赋值为1

  最后一步就是得到cur[]

1 for(int i=1;i<=m;i++)2 {3        for(int j=0;j<n;j++)4        {5              int num;6              scanf("%d",&num);7              if(num==0) cur[i] = (cur[i]|(1<<j));//这里要给0的地方变为1,1的地方放上0,因为要保证不合法的匹配一定是独一无二的。自己思考一下吧8        }9 }

技术分享

最后贴一下完整代码,一开始学的时候,感觉主流代码都一模一样,而且一大堆乱七八糟的函数,麻烦又看不懂,于是下定决心如果自己搞明白了,一定要写一个大家都看得懂的题解,感觉自己讲的比其他都清楚了,如果看不懂就真没办法了......

技术分享
 1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #define mod 100000000 5  6 using namespace std; 7 int dp[13][1<<12],cur[13]; 8 int can[1<<12],tot,m,n; 9 10 int main()11 {12     while(~scanf("%d%d",&m,&n))13     {14         tot = 0;15         for(int i=0;i<(1<<n);i++)16             if((i&(i<<1))==0) can[tot++] = i;17         memset(cur,0,sizeof(cur));18         memset(dp,0,sizeof(dp));19         for(int i=1;i<=m;i++)20         {21             for(int j=0;j<n;j++)22             {23                 int num;24                 scanf("%d",&num);25                 if(num==0) cur[i] = (cur[i]|(1<<j));26             }27         }28         for(int i=0;i<tot;i++)29             if((cur[1]&can[i])==0) dp[1][can[i]] = 1;30         for(int i=1;i<m;i++)31         {32             for(int j=0;j<tot;j++)33             {34                 if((can[j]&cur[i])==0)35                 {36                     for(int k=0;k<tot;k++)37                     {38                         if(((can[k]&cur[i+1])==0)&&((can[k]&can[j])==0))39                             dp[i+1][can[k]] = dp[i+1][can[k]]+dp[i][can[j]];40                     }41                 }42             }43         }44         int ans = 0;45         for(int i=0;i<tot;i++)46         {47             ans += dp[m][can[i]];48             ans = ans % mod;49         }50         printf("%d\n",ans);51     }52 }
View Code

 

【原创】【状态压缩DP】POJ3254 Corn Fields【新手向】