323:棋盘问题——DFS

描述

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放k个棋子的所有可行的摆放方案C。

输入

输入含有多组测试数据。
每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。

输出

对于每一组数据，给出一行输出，输出摆放的方案数目C （数据保证C<2^31）。

样例输入

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

样例输出

2
1

 

思路

搜索可取状态数可以用DFS也可以用BFS，但是DFS处理状态多的情况更好用。这里传参数的时候需要把遍历到第几行也传进去，因为如果不记录到第几行，就会出现顺序不一样，但是结果一样的重复情况。所以多传一个参，就能解决这个问题了！

 

#include <bits/stdc++.h>
using namespace std;

char ch[10][10];
int x[10],y[10],sum=0;

void dfs(int n,int k,int h){   //k剩余多少个棋子没有放
	if(k==0){
		sum++;
		return;
	}
	for(int i=h;i<=n;i++){
		if(x[i]==0){
			for(int j=1;j<=n;j++){
				if(y[j]==0&&ch[i][j]=='#'){
					ch[i][j]='.';
					x[i]=1;
					y[j]=1;
					dfs(n,k-1,i+1);   //已经遍历到i层，接下来从i+1层开始
					ch[i][j]='#';
					x[i]=0;
					y[j]=0;
				}
			}
		}
	}
	return;
}

int main(){
	int n,k;
	while(cin>>n>>k){
		if(n==-1&&k==-1) break;
		sum=0;
		memset(x,0,sizeof(x));
		memset(y,0,sizeof(y));
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				cin>>ch[i][j];
			}
		}
		dfs(n,k,1);
		cout<<sum<<endl;
	}
	return 0;
}