乍看上去还以为是个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;
}
···