互不侵犯king

T1T2
思路:DP+状压+预处理!
DP[i][s][k]为当前在i行,状态为s,放了k个国王的方案数!
DP[i][S][k]=∑ DP[i-1][s][k-f(s)] s是合法的上一层状态,f(p)是p的棋子数。
考虑每一次枚举,状态有2^9种=512,512*512*9*9*9 >1亿。(枚举2层状态,枚举行数,枚举更新棋子数,检查合法)。
貌似会超时吧!
于是准备预处理状态!
对于每行的状态,我们不能让一个国王的两侧有国王
先筛出符合的状态,记录在det数组中
枚举det数组中的一个状态,再枚举一个
判断两个状态作为上下行是否冲突,记录在一个布尔数组里即可

#include
#include
#include
#include
#include
#include
#include
#define LL long long
using namespace std;
LL n,k;
LL f[10][1<<11][90];//f[i][j][k]表示前i行  当前行状态j  棋子有k个 
LL gp[1<<11];//每个状态有几个棋子
LL sum;
bool map[1<<11][1<<11];//当前状态和上一层状态是否合法 
LL det[1<<11];
LL cnt=0;
void init()//预处理
{
    for(LL i=0;i<=sum;i++)
    {       
        if(!(i&(i<<1))&&!(i&(i>>1)))
        {
            det[++cnt]=i;
            for(LL j=0;jif(i&(1<for(LL i=1;i<=cnt;i++)
    for(LL j=i;j<=cnt;j++)
    {

        bool ok=true;
        if(det[i]&det[j])ok=false;
        if(det[i]&(det[j]<<1))ok=false;
        if(det[i]&(det[j]>>1))ok=false;
        if(ok)map[i][j]=map[j][i]=true;
    }
}
int main()
{
    scanf("%lld%lld",&n,&k);
    sum=(1<1;
    init();
    f[0][1][0]=1;
    for(LL i=0;ifor(LL j=1;j<=cnt;j++)
        {
            for(LL l=0;l<=k;l++)
            {
                if(f[i][j][l])
                {
                    for(LL p=1;p<=cnt;p++)
                    {
                        if(map[j][p]&&l+gp[p]<=k)
                        {
                            f[i+1][p][l+gp[p]]+=f[i][j][l];
                        }
                    }
                }
            }
        }
    }
    LL ans=0;
    for(LL i=1;i<=sum;i++)ans+=f[n][i][k];
    printf("%lld\n",ans);
    return 0;
}

你可能感兴趣的:(DP)