POJ - 1321简单搜索DFS

A - 棋盘问题

  POJ - 1321 

   

简单的搜索问题，类似于八皇后问题，只需要递归DFS求可能的数目，然后回溯即可。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n,k,cnt,tot;
char a[1000][1000];
int vis[1000];
void dfs(int cur){//行
  if (cnt == k){tot++;return;};//满足条件
  if (cur == n+1)return;//到达边界
  for(int i=1;i<=n;i++){ //遍历在cur(行)情况下的每一列情况
    if (a[cur][i]=='#' && !vis[i]){ //如果这一列有#并且这一列没有其他棋子
        cnt++;  
        vis[i]=1;  
        dfs(cur+1);//如果有可以放置的位置继续往下递归
        cnt--;//递归结束后这种情况已经讨论了，便跳过这种情况假设这种情况我们不放棋子
        vis[i]=0;
    }
  }
  dfs(cur+1); //如果没有找到继续向下一行找
}
int main(){
  while (~scanf("%d%d",&n,&k)){
    tot=0;
    cnt=0;
    memset(vis,0,sizeof(vis));
    if (n==-1 && k==-1)break;
    for (int i=1;i<=n;i++){
        for (int j=1;j<=n;j++){
           cin>>a[i][j];
        }
    }
    dfs(1);
    printf("%d\n",tot);

  }
  return 0;
}