POJ 1321 棋盘问题

题目链接：POJ 1321

题意：给一个n*n的棋盘和k颗棋子，只有棋盘上符号是'#'的位置才能放置棋子，且同行同列均最多只能有一颗棋子。问给出棋盘和k最多有多少种排法？

分析：

按行一行一行排放棋子，如果该行安排了棋子，那么将棋子所在列记标记，再在下一行安排下一颗棋子，直到将所有棋子安排完了，即所在行可以安排棋子，并且剩余棋子数为1.

注意点：

• 递归退出条件
• 还原每次递归完记得还原列状态
• 按行深搜，每次一个DFS

参考链接

CODE:

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 10
char pic[maxn][maxn],s[maxn];
int viscol[maxn];//viscol[i]表示第i列已经有棋子
int n, k, ans;

void DFS(int beginrow,int num)//从第beginrow行开始按行排列共num颗棋子，并把安排好的棋子所在列做标记
{
	for (int i = 0;i < n;i++)//这里的i表示列
	{
		if (pic[beginrow][i] == '#'&&viscol[i] == 0)
		{
			if (num == 1) ans++;//刚好排完
			else
			{
				viscol[i] = 1;
				for (int j = beginrow + 1;j <= n - num + 1;j++)//从第beginrow+1行开始排剩下的num-1颗棋子
				{
					DFS(j, num - 1);
				}
				viscol[i] = 0;
				//这次从beginrow行i列开始排第num颗棋子，应将第i列“释放”，为从beginrow行第i+1列开始排第num颗棋子做准备
			}
		}
	}
	
}

int main()
{
#ifdef LOCAL
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	while ((cin >> n >> k)&&n!=-1)
	{
		memset(viscol, 0, sizeof(viscol));
		ans = 0;
		for (int i = 0;i < n;i++)
		{
			cin >> s;
			for (int j = 0;j < n;j++)
				pic[i][j]=s[j];
		}
		for (int i = 0;i <= n - k;i++)
		{//由于每行最多只能有一颗棋子，要排k颗棋子的话，最多只能从第n-k行开始排（n-k到n-1共k行）
			DFS(i, k);
		}
		cout << ans << endl;
	}
	return 0;
}