kuangbin專題 專題一 簡單搜索 棋盤問題 POJ - 1321

 

題目連接：https://vjudge.net/problem/POJ-1321

 

題意：給一張棋盤，‘#’表示能夠下棋的地方，‘.’表示不能下棋的地方。
棋盤是n*n的，要求能放下k個棋子，要求k個棋子在不一樣行不一樣列

思路：dfs，首先遍歷地圖找到第一個能夠下棋的地方，而後從下一行開始繼續dfs，若是已下棋子數等於要求棋子數，答案++

---

 1 #include <iostream>
 2 #include <string.h>
 3 #include <algorithm>
 4 using namespace std;  5 
 6 #define rep(i,j,k) for(int i = (j); i <= (k); i++)
 7 #define per(i,j,k) for(int i = (j); i >= (k); i--)
 8 
 9 const int N = 15; 10 char mp[N][N]; 11 bool vis[N];//記錄列是否下的狀況，能夠看看紫書上八皇后的講解
12 int ans; 13 int n,k; 14 
15 void input(){ 16 
17     rep(i, 1, n){ 18         rep(j, 1, n) cin >> mp[i][j]; 19  } 20 } 21 
22 void init(){ 23     memset(vis, 0, sizeof(vis)); 24     ans = 0; 25 } 26 
27 void dfs(int cur,int chess){ 28 
29  rep(i, cur, n){ 30         if (!(n - i >= k - chess)) continue;//若是剩餘行數小於了剩餘還要下的棋子數，說明 31                                             //棋子沒法徹底下完直接忽略
32 
33         rep(j, 1, n){ 34 
35             if (vis[j] || mp[i][j] == '.') continue;//在這個dfs分支中j列已經有棋子 36                                                     //或者該空間沒法下棋子
37 
38             if (chess == k) { ans++; continue; } //棋子數知足了，ans++
39 
40             vis[j] = true;//標記該dfs分支j列有棋子
41             dfs(i + 1,chess + 1); 42             vis[j] = false;//回溯
43  } 44  } 45 } 46 
47 
48 int main(){ 49 
50     ios::sync_with_stdio(false); 51     cin.tie(0); 52 
53     while (cin >> n >> k){ 54         if (n == -1 && k == -1) break; 55 
56  init(); 57  input(); 58         dfs(1,1); 59 
60         cout << ans << endl; 61  } 62 
63 
64     return 0; 65 }