【复健系列】POJ 1321 棋盘问题 深度优先搜索

乍看上去还以为是个n皇后问题，然而并不是。
做DFS，控制好guard就好了。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<map>
#include<queue>
using namespace std;

int n,k;
char mmp[10][10];
bool visited[10];
int res = 0;
void search(int col,int kk){
    //printf("checking col %d with k %d\n",col,kk);
    //// GUARD: /////////////////////////////////////
    if(kk==0){  // 棋子用完了
        res++;
        return;
    }
    if(col>=n){ // 棋盘放完了
        return;
    }
    //// GUARD END //////////////////////////////////
    for(int i=0;i<n;i++){
        //// GUARD: /////////////////////////////////////
        if(visited[i])continue;        // 如果这一行放过了就不放
        if(mmp[i][col]=='.')continue;  // 如果这一行不是棋盘就不放
        //// GUARD END //////////////////////////////////
        //printf("thisstep: row %d col %d\n",i,col);
        visited[i]=true;
        search(col+1,kk-1);
        visited[i]=false;
    }
    search(col+1,kk);
}

int main(){
    while(~scanf("%d%d",&n,&k)){
        memset(visited,0,sizeof(visited));
        res = 0;
        if(n==-1||k==-1)return 0;
        for(int i=0;i<n;i++)scanf("%s",&mmp[i]);
        if(k == 0){
            printf("0\n");
            continue;
        }
        search(0,k);
        printf("%d\n",res);
    }
    return 0;
}
···