1896 互不侵犯 洛谷 luogu

题目描述

在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子。

注：数据有加强（2018/4/25）

输入输出格式

输入格式：

只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N

输出格式：

所得的方案数

------------------------------------------------------------------------------

懵

#include<iostream>
#include<cstdio>
using namespace std;
int n,k;
long long dp[][][]; //dp[i][j][k]表示第i行，状态为j，前面摆了k个国王时，方案数；
long long state[] , king[] ;//state[]是当前状态，king[]是当前行的国王数；
long long ans , sum;//ans是用来记录状态总数的，sum是用来计算一共有多少种方案的；

inline void inte()
{
    int tot = (<<n) - ;//最多到这个时候，就是二进制下，每一位上都放上国王，当然有不行的，为了方便下文排除；
    for(int i =  ; i <= tot ; i++)
        if(!((i<<)&i))         //因为要互不侵犯，所以，两个国王之间必须隔一个，这是判断是否满足题意国王之间不相互攻击；
        {
            state[++ans] = i;   //找到了满足的，记录这个状态；
            int t = i;
            while(t)                //判断这个状态有多少个国王，也就是t在二进制下有多少个1；
            {
                king[ans] += t%;
                t>>=;              //记住，是右移一位，和  t/=2 一样，就是稍微快一点；
            }
        }
}

int main()
{
    cin>>n>>k;                          //数据；
    inte();                             //初始化；
    for(int i = ; i <= ans ; i++)      //先处理第一行；
        if(king[i] <= k)                        //一行的国王数一定不能超过总数；
            dp[][i][king[i]] = ;

    for(int i =  ; i <= n ; i++)                           //处理剩下的，所以从 2 开始枚举；
        for(int j = ; j <= ans ; j++)                      //枚举状态；
            for(int p = ; p <= ans ; p++)                  //再一遍状态，用来当作上一行的状态，因为 我们由上向下递推，能迎上本行的，只有上一行；
            {
                //这里就不在赘述了，和处理第一行同理，但是不同的是这里处理相邻的行，
                if(state[j] & state[p]) continue;                //所以，上下相邻不行
                if(state[j] & (state[p]<<))    continue;        //本行的右上角不能有国王；
                if((state[j]<<) & state[p])    continue;       //左上角也不行；
                for(int s =  ; s <= k ; s++)
                {
                    //s表示本行以上用了多少国王； //满足条件后，还要记得国王数量是有限的！！
                    if(king[j] + s > k) continue;                //我们是递推，所以本行以上一定处理完了，所以，本行加以前用过的国王，总数不能超过限定；
                    dp[i][j][king[j]+s] += dp[i-][p][s];       //还记得dp[i][j][k]中的k表示已经用过的国王数，而king[]是本行的，s是本行以前的；
                }
            }

    for(int i = ; i <= n ; i++)                        //因为不确定在哪一行用光国王，所以都枚举一遍；
        for(int j =  ; j <= ans ; j++)
            sum += dp[i][j][k];                         //本行及以前用光了国王，那么方案数加在总数中；

    cout<<sum;
    return ;

}