kuangbin专题 专题一 简单搜索 棋盘问题 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 = ;
 char mp[N][N];
 bool vis[N];//记录列是否下的情况，可以看看紫书上八皇后的讲解
 int ans;
 int n,k;

 void input(){

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

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

 void dfs(int cur,int chess){

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

         rep(j, , n){

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

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

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

 int main(){

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

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

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

         cout << ans << endl;
     }

     return ;
 }