kuangbin专题 专题一 简单搜索 棋盘问题 POJ - 1321

棋盘问题 POJ - 1321

题目链接：https://vjudge.net/problem/POJ-1321

题意：给一张棋盘，‘#’表示可以下棋的地方，‘.’表示不能下棋的地方。
棋盘是n*n的，要求能放下k个棋子，要求k个棋子在不同行不同列

思路：dfs，首先遍历地图找到第一个可以下棋的地方，然后从下一行开始继续dfs，如果已下棋子数等于要求棋子数，答案++

#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;

#define rep(i,j,k) for(int i = (j); i <= (k); i++)
#define per(i,j,k) for(int i = (j); i >= (k); i--)

const int N = 15;
char mp[N][N];
bool vis[N];//记录列是否下的情况，可以看看紫书上八皇后的讲解
int ans;
int n,k;

void input(){

	rep(i, 1, n){
		rep(j, 1, n) cin >> mp[i][j];
	}
}

void init(){
	memset(vis, 0, sizeof(vis));
	ans = 0;
}

void dfs(int cur,int chess){

	rep(i, cur, n){
		if (!(n - i >= k - chess)) continue;//如果剩余行数小于了剩余还要下的棋子数，说明
											//棋子无法完全下完直接忽略

		rep(j, 1, n){

			if (vis[j] || mp[i][j] == '.') continue;//在这个dfs分支中j列已经有棋子
													//或者该空间无法下棋子

			if (chess == k) { ans++; continue; } //棋子数满足了，ans++

			vis[j] = true;//标记该dfs分支j列有棋子
			dfs(i + 1,chess + 1);
			vis[j] = false;//回溯
		}
	}
}


int main(){

	ios::sync_with_stdio(false);
	cin.tie(0);

	while (cin >> n >> k){
		if (n == -1 && k == -1) break;

		init();
		input();
		dfs(1,1);

		cout << ans << endl;
	}


	return 0;
}