Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放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
刚开始想的是在一个放棋子的地方可以有(n-行)种情况,后来发现想错了,参考别人的代码;棋子都是想同的,摆完第一行接着摆第二行,摆到最后sum++,dfs所有情况,最后输出;
注意:(1)用来标记的数组设为一位数组更简单,因为题目要求不是同一列,可以放心用一维数组;
(2)还有就是k < n时,注意有的行可以不放棋子,即在for循环之后再加一行dfs(row+1)表示这行不放棋子。
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
using namespace std;
int n,k,res_k = 0;
char chess[10][10];
int visit[10];
long long sum = 0;
void dfs(int row)//行
{
if(res_k == k)
{
sum++;
return;
}
if(row >= n)//搜出棋盘外
return;
for(int i = 0; i < n; i++)
{
if(chess[row][i] == '#' && visit[i] == 0)//不同列,罗列在一起就是一行 <= n个1
{
visit[i] = 1;
res_k++;
dfs(row + 1);
visit[i] = 0;
res_k--;
}
}
dfs(row + 1);//row行不放棋子,进行下一行判断
}
int main()
{
while((cin >> n >> k) && n != -1)
{
res_k = 0;//棋子
sum = 0;//摆放次数
memset(visit,0,sizeof(visit));
for(int i = 0; i < n; i++)
cin >> chess[i];
dfs(0);
cout << sum << endl;
}
return 0;
}