状压DP入门

洛谷题号p1896
在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子。
题解：
首先暴搜可肯定是超时的所以我门考虑状压 因为每个数都可以用二进制表示出来 二进制中01可以表示当前行放的棋子的位置 ，以及个数 首先我们定义一个dp数组dp[i][j][x]表示第i行放的是j的二进制位数的形式前i一共放了x个。
首先我们定义一个数组判断当前行是否合法 首先我们引入&操作 这个操作表示两个数二进制相互&操作 要是都为1的话则为1有一个为0的话则为0.可以用户判断当前行是否合法主要代码如下

  bool  check1(int x)
  {
  for(int i=0;i+1<n;i++)
    if((x&(1<<i))&&(x&(1<<(i+1)))) return false; // 要是有格连续放为1即为不合法
  return true;
  }

其次 我们在定义一个函数用来判断当前行和上一行的棋子放置位置的状态是否合法 主要是判断要是当前位置放置了棋子 其左上角和右上角不能放置棋子
主要代码如下

bool check(int st1,int st2)
{
    for(int i=0;i<n;i++)
    {
       if(st1&(1<<i)) // 当前st2 前一行为st1
       {
          if(st2&(1<<i))return false;
          else if(i+1<n&&(st2&(1<<(i+1))))return false;
          else if(i-1<n&&(st2&(1<<(i-1)))) return false;
       }
    }
return true;
}

判断二进制位数代码如下

   int c(int x)
   {
   int num=0;
   while(x)
   {
   if(x&1)num++;
   x/=2;
   }
   return num;
   }

整体实现代码如下

#include<bits/stdc++.h>
using namespace std;
int n,k;
long long  dp[10][1<<9][88];//从左往右依次表示第i行放置情况为j的二进制情况
//最后一位表示前i行一共放了多少个棋子
int c(int x)
{
    int num=0;
    while(x)
    {
        if(x&1)num++;
        x/=2;
    }
    return num;
}
bool check1(int x)//分析该行是否合法
{
    for(int i=0;i+1<n;i++)
    {
        if((x&(1<<i))&&(x&(1<<(i+1))))return false;
    }
    return true;
}

bool check2(int x,int x1)//判断当前行和前一行是否合法
{
    for(int i=0;i<n;i++)
    {
        if(x&(1<<i))
        {
            if(x1&(1<<i)) return false;
            else if(i+1<n&&(x1&(1<<(i+1))))return false;
            else if(i-1<n&&(x1&(1<<(i-1))))return false;
        }
    }
    return true;
}
int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        for(int st=0;st<(1<<n);st++)
        {
            if(!check1(st))continue;
            if(i==0)dp[i][st][c(st)]=1;//第0行合法都可设置为1 
            else
            {
              for(int j=c(st);j<=k;j++) //表示放j个
              {
                   for(int st2=0;st2<(1<<n);st2++)
                   {
                       if(!check1(st2)||!check2(st,st2)) continue;
                       dp[i][st][j]+=dp[i-1][st2][j-c(st)];
                   }
              }
            }
        }
    }
      long long ans=0;
      for(int i=0;i<(1<<n);i++)
        ans+=dp[n-1][i][k];
        printf("%lld\n",ans);
        return 0;
}