【ACWing】1114. 棋盘问题

题目地址：

https://www.acwing.com/problem/content/description/1116/

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放$k$个棋子的所有可行的摆放方案数目$C$。

输入格式：
输入含有多组测试数据。
每组数据的第一行是两个正整数$n,k$，用一个空格隔开，表示了将在一个$n\ast n$的矩阵内描述棋盘，以及摆放棋子的数目。当为-1 -1时表示输入结束。随后的$n$行描述了棋盘的形状：每行有$n$个字符，其中#表示棋盘区域，.表示空白区域（数据保证不出现多余的空白行或者空白列）。

输出格式：
对于每一组数据，给出一行输出，输出摆放的方案数目$C$（数据保证$C<2^{31}$）。

数据范围：
$n\le 8,k\le n$

思路是DFS。DFS每一层代表行，每一行枚举填充哪一列（当然也可以不填充这一行）。如果填满了$k$个棋子，则得到一个方案。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 10;
char g[N][N];
int n, k;
bool col[N];

// dfs返回从第x行开始填，已经填过cnt个棋子的情况下，有多少个方案。
int dfs(int x, int cnt) {
  if (cnt == k) return 1;

  if (x > n) return 0;
  int res = 0;
  for (int i = 1; i <= n; i++)
    if (g[x][i] == '#' && !col[i]) {
      cnt++;
      col[i] = true;
      res += dfs(x + 1, cnt);
      cnt--;
      col[i] = false;
    }
  // 还需要考虑第x行不填的情况
  res += dfs(x + 1, cnt);
  return res;
}

int main() {
  while (scanf("%d%d", &n, &k)) {
    if (!~n && !~k) break;
    memset(col, 0, sizeof col);
    for (int i = 1; i <= n; i++) scanf("%s", g[i] + 1);
    printf("%d\n", dfs(1, 0));
  }
}

时间复杂度$O(n!)$，空间$O(n)$。