洛谷P1896 互不侵犯[SCOI2005]【状压DP】

题目描述 https://www.luogu.org/problemnew/show/P1896
这是一道比较典型的状压DP题，只是这题国王的攻击范围为8个格子，所以判断的时候不光本身要左移判断，本身也要与下一行的左移判断（右移一样），同时这个题也有国王的数目要求，所以转移的数组要加上一个变量，即
ans[i][p][k]表示第i行 状态为p 已经放了k个国王
转移的时候还要注意国王总数不能超过k

优化
1.将合法的状态存入数组中，减少循环次数
2.预处理两个状态挨着是否合法

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
ll ans[10][1500][100],num[1500],sta[1500],an,cnt;
bool fl[1500][1500];
int main()
{ ll n,k; 

   scanf("%lld%lld",&n,&k);
   for(ll i=0;i<=(1<<n)-1;i++)
     if((i&(i<<1))==0&&(i&(i>>1))==0)//判断自身合法 
     {
     	sta[++cnt]=i;
     	for(ll j=0;j<=n-1;j++)
          if((i&(1<<j))!=0) num[cnt]++;//这个状态的国王数 
	 }
   for(ll i=1;i<=cnt;i++)
     for(ll j=i;j<=cnt;j++)
	 {
	 	ll x1=sta[i],x2=sta[j];
	 	if((x1&x2)!=0||(x1&(x2<<1))!=0||(x1&(x2>>1))!=0) continue;
	 	fl[i][j]=fl[j][i]=1;
	 }//枚举两个状态 判断两个状态之间是非合法 
   ans[0][1][0]=1;//ans[i][p][k]表示第i行 第p个状态 已经放了k个国王 
   for(ll i=0;i<=n-1;i++)//枚举行 
     for(ll j=1;j<=cnt;j++)//枚举第几个状态 
       for(ll l=0;l<=k;l++)//枚举国王数 
         for(ll p=1;p<=cnt;p++)//枚举下一行第几个状态
           if(fl[j][p]==1&&num[p]+l<=k)
             ans[i+1][p][num[p]+l]+=ans[i][j][l];
   for(ll i=1;i<=cnt;i++)//答案为最后一行状态相加                 
	 an+=ans[n][i][k];
   printf("%lld",an); 
 
  return 0;	
}