棋盘问题
问题描述
小H收集到一些形状特殊的棋盘,她想在棋盘上面摆放棋子(棋子都是相同的)。她希望摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,你能帮她求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案数C嘛?
输入格式
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
注意只有#棋盘区域可以摆放棋子。
输出格式
对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)
样例输入
2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1
样例输出
2
1
数据规模和约定
1<=k<=n<=8
思路:
使用dfs
在dfs函数中,有四个参数,k
表示当前放置棋子数,K
表示最多放置棋子数,a
表示当前判断要不要放棋子的那一行的行下标,n
表示棋盘总行数。
当放置棋子数等于最大放置棋子数时就停止放置。
当判断放置棋子的行下标超出棋盘行数时停止。
由于把行下标作为迭代的参数放在函数的传入参数中,因此要判断不放棋子的情况,否则会出现遍历完当前行之后,因选过放棋子后继续遍历当前行,但是无法进入下一行而停止dfs的情况。(这里需要假设推理一下,这一步是我在写代码时错误的地方
),比如情况如下:
#.
.#
此时如果没有第20行dfs(k, K, a + 1, n);放弃放棋子的选择,则在第一个#完成一次放置后无法进入第二行,使得遍历不到右下角的#
注意:最好直接在全局声明数组,开大的空间。
注意在输入字符串时,最好使用string进行输入,再使用这个string的变量赋值给字符串变量,免得把回车也赋值进去
#include <bits/stdc++.h>
#include <iostream>
#include <vector>
using namespace std;
long long c = 0;
char qipan[9][9];
int v[9] = {0};
//数组v表示列上有无棋子,qipan表示棋盘信息,k表示当前放置的棋子数量,a表示放的放的行,为避免重复计算而设,n表示棋盘大小
void dfs(int k, int K, int a, int n)
{
if (k == K)
{
c++;
return;
}
if (a == n) //此时表示虽然到了最后一步但是还没到放置棋子数量还没到k的情况
{
return;
}
dfs(k, K, a + 1, n);//这行选择不放棋子,直接进下一行判断
for (int i = 0; i < n; i++)//若选择放棋子,则遍历这一行的每一列,因为每一行行数不同,所以不会出现124和421这种重复的情况
{
if (qipan[a][i] == '#')
{
if (v[i] != 1)
{
v[i] = 1;
dfs(k + 1, K, a + 1, n);
v[i] = 0;
}
}
}
}
int main()
{
while (true)
{
int n, k;
cin >> n >> k;
if (n == -1 && k == -1)
break;
for (int i = 0; i < n; i++)
{
string s;
cin >> s;
for (int j = 0; j < s.size(); j++)
{
qipan[i][j] = s[j];
}
}
vector<int> v(n, 0);
dfs(0, k, 0, n);
cout << c << endl;
c = 0;
}
}