棋盘问题 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;
}