virtual Judge-棋盘问题题解

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放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;
}