* 深度優先搜索(1) 編程題#1: 棋盤問題(Coursera 程序設計與算法 專項課程4 算法基礎;遞歸+剪枝!)

編程題#1: 棋盤問題

來源: POJ (http://bailian.openjudge.cn/practice/1321/)

注意: 總時間限制: 1000ms 內存限制: 65536kB

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

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

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

樣例輸入

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

樣例輸出

2
1

典型深搜題!
解題思路:
按行循環即可,因爲一行只能放一個棋子!每行有放棋子和不放棋子兩種情況!

#include <iostream>
#include <cstring>
using namespace std;

int programs;  //方案數(全局變量)
char chessboard[9][9];
int chessUsedY[9];
int n, k;

void dfs(int row, int kk){  //row表示第row行,kk表示剩餘棋子數
    if (n - row + 1 < kk)  //剪枝!剩餘行數小於棋子數,方案失敗
        return;
    if (row > n && kk > 0)   //失敗方案。行數row已經越界,可能出現這種情況的,因爲有的行沒有放棋子!
        return;
    if (kk == 0){
        programs++;
        return;
    }

    for (int i = 1; i <= n; i++){  
        if (chessUsedY[i] == 0 && chessboard[row][i] == '#'){
            chessUsedY[i] = 1;
            dfs(row + 1, kk - 1);  //第row行放棋子的情況,此時棋子未放完,繼續遞歸深搜
            chessUsedY[i] = 0;  //調用上一個dfs(row + 1, kk - 1)函數時chessUsedY[i] = 1;但是調用結束後chessUsedY[i] = 0;因爲其它函數可能經過這一路徑!
        }
    }

    dfs(row + 1, kk);      //第row行不放棋子的情況
}

int main(){
    while (cin >> n >> k){
        if (n == -1 && k == -1)
            break;
        programs = 0;
        memset(chessUsedY, 0, sizeof(chessUsedY));  //未使用的區域爲0,使用過的區域爲1

        for (int i = 1; i <= n; i++){
            for (int j = 1; j <= n; j++){
                cin >> chessboard[i][j];
            }
        }

        dfs(1, k);

        cout << programs << endl;
    }

    return 0;
}

其它解法參考:
http://blog.csdn.net/qq_35083093/article/details/70598133

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章