poj1321 棋盘问题
dfs
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
主要想法
dfs的基本应用,因为每行和每列是不能有两颗棋子的,所以标记第几列是否有棋子就可以了。然后搜索每一行满足条件的。注意满足条件的位置是可以不放棋子的,所以要考虑当前位置不放子的情况。
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,k;
int maze[10][10];
int col[10];
long long out = 0;
int k_num = 0;
int dfs(int num,int k_num)
{
if(k_num == k)
{
out++;return 0;
}
if(num == n)
{return 0;}
for(int i = 0;i < n; i++)
if(!col[i] && !maze[num][i])
{
col[i] = 1;
dfs(num+1,k_num+1);
col[i] = 0;
}
dfs(num+1,k_num);//当前位置不放子
}
int main()
{
while(scanf("%d %d",&n,&k) != EOF && n != -1)
{
for(int i = 0; i < n; i++)
{
char data[15];
scanf("%s",data);
for(int j = 0; j < n ; j++)
{
if(data[j] == '#') maze[i][j] = 0;
else maze[i][j] = 1;
}
}
out = 0;k_num = 0;
memset(col,0,sizeof(col));
dfs(0,0);
printf("%I64d\n",out);
}
return 0;
}
真的是感觉处理成0和1比读字符方便多了,不需要考虑回车怎么办。
还有这个的数据2^31肯定是没int大的,虽然我也没有试改成int会不会出问题。