**********此文章属于原创,看此文章前请先参考论文 周伟《动态规划之状态压缩》**********
问题1:
在n*n(n≤20)的方格棋盘上放置n个车(可以攻击所在行、列),求使它们不能互相攻击的方案总数。
如果用组合学的角度来考虑此问题,那么非常简单:
我们一行一行放置,第一行有n种选择,第二行n-1,……,最后一行有1种选择,根据乘法原理,答案就是n!
这里我们介绍另一种解法:状态压缩递推(States Compressing Recursion,SCR)。
- 我们仍然一行一行放置。
- 取棋子的放置情况作为状态,某一列如果已经放置棋子则为1,否则为0。这样,一个状态就可以用一个最多20位的二进制数表示。
- 例如n=5,第1、3、4列已经放置,则这个状态可以表示为01101(从右到左)。设fs为达到状态s的方案数,则可以尝试建立f的递推关系。
- 考虑n=5,s=01101
- 因为我们是一行一行放置的,所以当达到s时已经放到了第三行。又因为一行能且仅能放置一个车,所以我们知道状态s一定来自:
①前两行在第3、4列放置了棋子(不考虑顺序,下同),第三行在第1列放置;
②前两行在第1、4列放置了棋子,第三行在第3列放置;
③前两行在第1、3列放置了棋子,第三行在第4列放置。
这三种情况互不相交,且只可能有这三种情况,根据加法原理,fs应该等于这三种情况的和。写成递推式就 是:
f(01101) = f(01100) + f(01001) + f(00101);
- 根据上面的讨论思路推广之,得到引例的解决办法:
f(0) = 1;
f(s) = ·f(s-2^i);
其中s的右起第i+1位为1(其实就是在枚举s的二进制表示中的1)
代码如下:
#include<iostream>
#include<cstdio>
using namespace std;
long long f[1<<20];
int main(){
long long n;
while(cin>>n){
memset(f,0,sizeof(f));
f[0] = 1;
long long i,t;
for( i=1; i< 1<<n; i++){
for( t=i; t>0; t -= (t & -t)){
f[i] += f[i & ~(t & -t)]; //注意理解
}
}
cout<<f