在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
Output
对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。
Sample Input
2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1
Sample Output
2
1
题解
一开始总觉得题目怪怪的,其实就是只能在#内放棋子。然后就是题目的要求了,放的棋子它的行列都不能有其他棋子,就是简化了的八皇后问题。
思路跟八皇后问题差不多,就是利用dfs一步一步去试验,不行就回溯。值得注意的是,八皇后问题中,n x n的矩阵应放n个皇后,而这题的棋子是可以小于矩阵的。
#include<iostream>
#include<string>
using namespace std;
char map[10][10]; //输入的棋盘
bool vis[10]; //标记数组 表示该列有没有放棋子
int n, k,ans,sum; //ans是输出的方案数目 sum是已放的棋子的个数
void dfs(int step) //step为步数也为行数
{
if (sum == k) //当已放棋子数等于需要放的棋子数
{
ans++; //总数加1
return; //返回 重新开始另一种方案
}
if (step >= n)return; //行数大于或等于矩阵行数 返回
for (int i = 0; i < n; i++) //从当前行的0列开始遍历到最后一列
{
if (map[step][i]=='#' && vis[i]==0 ) //判断能否放棋子
{
sum++; //能放棋子 已放棋子数+1
vis[i] = 1; //赋值为1 表示该列已放棋子 不能再放了
dfs(step+1); //下一行
sum--; //回溯 已放棋子数-1
vis[i] = 0; //重新将标记置0
}
}
dfs(step + 1); //下一行
//如果没有该语句,当输入的棋子数比行数小时,递归每次调用到等于棋子数时就结束了,无法进行下一行
//如样例1
}
int main()
{
int i, j; //循环用的变量
while (cin >> n >> k) //输入
{
ans = 0;sum = 0; //每次数据输入后都不要忘记将这两个变量重新赋0
if (n == -1 && k == -1)break; //遇到-1 -1退出
for (i = 0; i < n; i++)
{
for (j = 0; j < n; j++)
cin >> map[i][j];
}
memset(vis, 0,sizeof(vis)); //将标记数组全部置0
//虽然定义全局变量会默认全0 但第二次的数据输入不会 因此需要此操作
dfs(0); //调用dfs
cout << ans << endl; //输出结果
}
//system("pause");
return 0;
}