状压dp 互不侵犯

这是一道典型的状压dp题。
我们可以将每一行棋子的状态用二进制存储。 dp[i][j][k]表示到第i行第j种状态已有k个棋子。
那么状态转移方程为dp[i+1][p][k+num[p]]+=dp[i][j][k] (p指下一行的状态)。
我们可以用位运算来判断状态是否合法，上下棋子是否攻击到。
上代码

#include<iostream>
#include<cstdio>
using namespace std;
long long dp[15][1024][90];
long long n,k,cnt,st[1030];
long long ans,n1[1030];
bool ok[1024][1024];
long long num(long long x)
{
	long long sum=0;
	while(x>0)
	{
		int y;
		y=x%2;
		if(y) sum++;
		x=x/2;
	}
	return sum;
}
void gg()
{
	for(long long i=0;i<=(1<<n)-1;i++)
	{
		long long x1=i>>1,x2=i<<1;
		if(x1&i) continue;
		if(x2&i) continue;
		cnt++;st[cnt]=i;n1[cnt]=num(i);
	}
	for(long long i=1;i<=cnt;i++)
	 for(long long j=1;j<=cnt;j++)
	 {
	 	long long x1=st[j]>>1,x2=st[j]<<1;
	 	if(st[i]&st[j]) continue;
	 	if(st[i]&x1) continue;
	 	if(st[i]&x2) continue;
	 	ok[i][j]=ok[j][i]=1;
	 }
}
int main()
{
	long long i,j,l,p;
	cin>>n>>k;
	gg();
	dp[0][1][0]=1;
	for(i=0;i<=n-1;i++)
	 for(j=1;j<=cnt;j++)
	  for(l=0;l<=k;l++)
	   for(p=1;p<=cnt;p++)
	   {
	   	if(!ok[j][p]) continue;
	   	if(l+n1[p]>k) continue;
	   	dp[i+1][p][l+n1[p]]+=dp[i][j][l];
	   }
     for(j=1;j<=cnt;j++)
     {
     	ans+=dp[n][j][k];
      }
	 cout<<ans;
	 return 0;
}