状压DP是一种在与选集合有关的DP,一般借助二进制枚举状态。
下面是典型例题:
例题1
这题如果直接搜索,复杂度会很高。经过观察发现一旦某一行选好,那么下一行所有合法的状态就能表示出来。
可以借助用二进制数[0…2m)表示某一行的所有状态,其中m为列数。(举例:二进制数00101表示选了第0和第2列)。对于任意一个状态t,若t&(t>>1)不为0则表明行内选了相邻的,是不合法状态;若它与上一行另一个状态k满足j&k不为0说明选择了两行之间有相邻的,也是不合法的。
另外,如果已确定状态可能合法,那么还要枚举是否选择了不可选的0。
优化:不难发现当t>=(1<<(m-1))+(1<<(m-2)后便再无合法状态,因此不用枚举所有2m种状态。
实现如下:
for(int i=1; i<=n; i++) {
for(int j=0; j<m; j++) {
if(!check(i,j)) //行内相交或选了0
continue;
for(int k=0; k<m; k++) {
if(check(i-1,k)&&!(j&k)) {
dp[i][j]=(dp[i][j]+dp[i-1][k])%MOD;
}
}
}
int res=0;
for(int i=0; i<m; i++) {
res=(res+dp[n][i])%MOD;
}
printf("%d",res);
}
例题2
此题之前用搜索做过,由于与上一题相似,因此用状压DP再做一遍。
具体细节见代码:
bool noCommon(uInt j,uInt k) {
return ((j&k)==0)&&((j&(k>>1))==0)&&((j&(k<<1))==0);
}
uInt solve(int n,int m) {
if(m<2) m=2;
else m=(1<<(m-1))+(1<<(m-2));
for(int i=1; i<=n; i++) {
for(int j=0; j<m; j++) {
if((j&(j>>1))!=0) continue;
for(int k=0; k<m; k++) {
if((k&(k>>1))!=0) continue;
if(noCommon(k,j)) {
dp[i][j]=max(dp[i][j],dp[i-1][k]+calc(i,j));
}
}
}
}
return *max_element(dp[n],dp[n]+m);
}