POJ1321——DFS——棋盘问题
Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
每组数据的第一行是两个正整数,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
Source
大意:注意每一次DFS后的回溯把改变的值变回去,用cin比scanf在字符型的输入中好多了!
#include<cstring> #include<cstdio> #include<algorithm> #include<iostream> using namespace std; int n,k,ans; bool map[10]; bool mark[10][10]; void dfs(int col ,int m) { if(m == k){ ans++; return; } if(col > n) return ; for(int j = 1; j <= n ;j++){ if(mark[col][j] == true && map[j] == false){ map[j] = true; dfs(col+1,m+1); map[j] = false; } } dfs(col+1,m); return ; } int main() { char temp; while(cin >> n >> k){ if(n == -1 && k == -1) break; ans = 0; memset(mark,false,sizeof(mark)); memset(map,false,sizeof(map)); for(int i = 1; i <= n ;i++){ for(int j = 1; j <= n ;j++){ cin >> temp; if(temp == '#') mark[i][j] = true; } } dfs(1,0); printf("%d\n",ans); } return 0; }