题目地址:
https://www.acwing.com/problem/content/description/1116/
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放 k k k个棋子的所有可行的摆放方案数目 C C C。
输入格式:
输入含有多组测试数据。
每组数据的第一行是两个正整数
n
,
k
n,k
n,k,用一个空格隔开,表示了将在一个
n
∗
n
n∗n
n∗n的矩阵内描述棋盘,以及摆放棋子的数目。当为-1 -1
时表示输入结束。随后的
n
n
n行描述了棋盘的形状:每行有
n
n
n个字符,其中#
表示棋盘区域,.
表示空白区域(数据保证不出现多余的空白行或者空白列)。
输出格式:
对于每一组数据,给出一行输出,输出摆放的方案数目
C
C
C(数据保证
C
<
2
31
C<2^{31}
C<231)。
数据范围:
n
≤
8
,
k
≤
n
n≤8,k≤n
n≤8,k≤n
思路是DFS。DFS每一层代表行,每一行枚举填充哪一列(当然也可以不填充这一行)。如果填满了 k k 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!) O(n!),空间 O ( n ) O(n) O(n)。