#10170. 「一本通 5.4 例 1」国王

题目描述

原题题面

在n*n 的棋盘上放k 个国王，国王可攻击相邻的 8个格子，求使它们无法互相攻击的方案总数。

输入格式

只有一行，包含两个整数 n和k。

输出格式

每组数据一行为方案总数，若不能够放置则输出 0。

样例输入 1

3 2

样例输出 1

16

样例输入 2

4 4

样例输出 2

79

【解题思路】：

%一下这位dalao
思路和注释都很清晰，好文章，%%%

【AC代码】:

#include 
#define M(a, b) memset(a, b, sizeof(a))
#define INF 0x3f3f3f3f
#define MOD 10000007
using namespace std;
long long f[11][205][205], ans = 0;
int con[205], s[205];
int N, K, i, j, k, s0;

void work1() {
    s0 = 0;
    int k, i, j;
    for (i = 0; i < (1 << N); ++i) {
        if (i & (i << 1))
            continue;
        k = 0;
        for (j = 0; j < N; ++j)
            if (i & (1 << j))
                ++k;
        s[++s0] = i;
        con[s0] = k;
    }
}

void work2() {
    f[0][1][0] = 1;
    for (i = 1; i <= N; i++) {
        for (j = 1; j <= s0; j++) {
            for (k = 0; k <= K; k++) {
                if (k >= con[j]) {
                    for (int x = 1; x <= s0; x++) {
                        if (!(s[x] & s[j]) && !(s[x] & (s[j] << 1)) && !(s[x] & (s[j] >> 1))) {
                            f[i][j][k] += f[i - 1][x][k - con[j]];
                        }
                    }
                }
            }
        }
    }
    for (i = 1; i <= s0; i++) ans += f[N][i][K];
    printf("%lld\n", ans);
}

int main() {
    scanf("%d %d", &N, &K);
    work1();
    work2();
    return 0;
}