状压dp:在我们解决动态规划题目的时候,dp数组最重要的一维就是保存状态信息,但是有些题目它的具有dp的特性,并且状态较多,如果直接保存的可能需要三维甚至多维数组,这样在题目允许的内存下势必是开不下的,那么我们能不能想个办法,把它压缩成一维呢?对,二进制.一般的动规题目数据范围都不会太大,那么就可以把几个状态全部压缩成一个二进制数保存下来,这样就大大节省了空间,来允许我们进行其他的操作,这就叫做状态压缩.运用状态压缩来保存状态的dp就叫做状压dp,这类dp一般数据范围有一项很小(好像是不超过16吧),看到这种数据范围就可以往状压上想。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
long long i,j,kk,p,n,k,dp[10][2000][100],dge[500][500],sta[1000];
long long num[1000];
long long ans=0;
long long cnt;
void jud() {
long long s;
for(s=0;s<=(1<<n)-1;s++) {
if((!((s<<1)&s))&&(!((s>>1)&s))) {
sta[++cnt]=s;
for(int i=0;i<n;i++) if(((1<<i)&s)) num[cnt]++;}
}
//枚举一行里的状态是否可行。用sta存状态,用num存此状态下有多少个1即用了多少个。(1<<i)&s用来逐位判断是否有1
for(i=1;i<=cnt;i++)
for(j=i;j<=cnt;j++) {
int s1=sta[i],s2=sta[j];
if((s1&s2)||(s1&(s2<<1))||(s1&(s2>>1))) continue;
dge[i][j]=dge[j][i]=1;
}
//判断上下两行上行sta【i】下行sta【j】是否可行。用dge存是否可行
}
int main()
{
scanf("%ld%ld",&n,&kk);
jud();
dp[0][1][0]=1;
//dp【i】【j】【k】i行第j种所有共放k个有几种可能
for(i=0;i<=n-1;i++)
for(j=1;j<=cnt;j++) {
for(k=0;k<=kk;k++)
for(p=1;p<=cnt;p++) {
if(!dge[j][p]||(k+num[p]>kk)) continue;
dp[i+1][p][k+num[p]]+=dp[i][j][k];
}
}
for(i=1;i<=cnt;i++)
ans+=dp[n][i][kk];
printf("%ld\n",ans);
return 0;
}