POJ 1321 棋盤問題 簡單的DFS

棋盤問題

Time Limit:1000MS     Memory Limit:10000KB     64bit IO Format:%I64d & %I64u

Description

在一個給定形狀的棋盤（形狀可能是不規則的）上面擺放棋子，棋子沒有區別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列，請編程求解對於給定形狀和大小的棋盤，擺放k個棋子的所有可行的擺放方案C。

Input

輸入含有多組測試數據。 
每組數據的第一行是兩個正整數，n k，用一個空格隔開，表示了將在一個n*n的矩陣內描述棋盤，以及擺放棋子的數目。 n <= 8 , k <= n 
當爲-1 -1時表示輸入結束。 
隨後的n行描述了棋盤的形狀：每行有n個字符，其中 # 表示棋盤區域， . 表示空白區域（數據保證不出現多餘的空白行或者空白列）。 

Output

對於每一組數據，給出一行輸出，輸出擺放的方案數目C （數據保證C<2^31）。

Sample Input

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

Sample Output

2
1

解題思路：

一行一行的看，每行肯定只能放一個，有點類似於八皇后。

在標記上我用了位運算來標記當前列是否被佔用。

我把所有空位置爲-1。剩下的就是暴力搜。

AC代碼：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cstdlib>
using namespace std;

const int maxn = 8;
int vis[maxn][maxn];
int n, k, ans;

void DFS(int cnt,int row,int line){
    for(; line < n; line++){
        for(int j = 0; j < n; j++){
            if(vis[line][j] == -1)continue;
            int t = 1 << j;  //當前位置的行數置1
            if(t&row) continue; //如果與標記行數同爲1，則說明該行已標記
            if(cnt+1 == k){
                ans++;
                continue;
            }
            DFS(cnt+1,row|t,line+1);   //row|t  把當前行標記一下。
        }
    }
}

int main(){
    while(cin >> n >> k){
        if(n == -1 && k == -1)break;
        memset(vis,0,sizeof(vis));
        for(int i = 0; i < n*n; i++){
            char ch;
            cin >> ch;
            if(ch == '.') vis[i/n][i%n] = -1;
        }
        ans = 0;
        DFS(0,0,0);
        cout << ans << endl;
    }
}