标签：洛谷 int res P1896 STATES MAXN SCOI2005 include dp

链接:https://www.luogu.org/problemnew/show/P1896

题意:

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

思路：

状压dp，dp[i][j][k]。表示。第i行，放j个国王，第j种状态的方法。

预处理第一行，dp。

枚举当前行和上一行的状态。

代码：

#include <iostream>
#include <memory.h>
#include <string>
#include <istream>
#include <sstream>
#include <vector>
#include <stack>
#include <algorithm>
#include <map>
#include <queue>
#include <math.h>
#include <cstdio>
using namespace std;

typedef long long LL;

const int MAXN = 10;
const int STATES = 400;

int states[STATES];
int num[STATES];
LL dp[MAXN][MAXN * MAXN][STATES];
int pos = 0;

int Get_w(int n)
{
    int res = 0;
    while (n)
    {
        if (n & 1)
            res++;
        n >>= 1;
    }
    return res;
}

void Init(int w)
{
    int total = (1 << w);
    for (int i = 0;i < total;i++)
    {
        if ((i & (i << 1)) == 0)
        {
            states[++pos] = i;
            num[i] = Get_w(i);
        }
    }
}

int main()
{
    int n, kk;
    scanf("%d%d", &n, &kk);
    Init(n);
    for (int i = 1;i <= pos;i++)
        dp[1][num[states[i]]][states[i]] = 1;
    for (int i = 2;i <= n;i++)
    {
        for (int j = 1;j <= pos;j++)//当前行
        {
            for (int k = 1;k <= pos;k++)//上一行
            {
                if (states[j] & states[k] || (states[j] << 1) & states[k] || (states[j] >> 1) & states[k])
                    continue;
                for (int x = 0;x <= kk - num[states[j]];x++)
                        dp[i][num[states[j]] + x][states[j]] += dp[i - 1][x][states[k]];
            }
        }
    }

    LL res = 0;
    for (int i = 1;i <= pos;i++)
        res += dp[n][kk][states[i]];
    printf("%lld\n", res);

    return 0;
}

标签：洛谷,int,res,P1896,STATES,MAXN,SCOI2005,include,dp
来源： https://www.cnblogs.com/YDDDD/p/10360037.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。