状压dp——洛谷 P1377 M国王

https://www.luogu.org/problem/show?pid=1377#sub

状压dp。rnf[i][j][k]表示前i行放j个,第i行的放置情况为k的方案数。
转移:f[i][j][k]=sum(f[i-1][j-num[k]][k’])(k’和k的相同位和相邻位不都是1,用位运算判断)r
num[k]表示k这种状态的1的个数
最后输出sum(f[n][m][k]),k为任意状态

其实很水诶……

#include
#define Ll long long
using namespace std;
int num[500],w;
int f[10][100][500];
int n,m,ans;
int main()
{
    scanf("%d%d",&n,&m);
    w=(1<1;
    for(int i=0;i<=w;i++)
        for(int k=i;k;k>>=1)if(k&1)num[i]++;
    f[0][0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=0;j<=m;j++)
            for(int k=0;k<=w;k++)if(!((k<<1)&k)&&!((k>>1)&k))
                if(j>=num[k])
                    for(int kk=0;kk<=w;kk++)if(!((kk<<1)&kk)&&!((kk>>1)&kk))
                        if(!(kk&k)&&!((k<<1)&kk)&&!((k>>1)&kk))
                            f[i][j][k]+=f[i-1][j-num[k]][kk];
    for(int k=0;k<=w;k++)ans+=f[n][m][k];
    printf("%d",ans);
}

你可能感兴趣的:(状压dp)