POJ.1321 棋盘问题 （DFS）

POJ.1321 棋盘问题 （DFS）

题意分析

给出n*n的棋盘，棋盘中#表棋盘区域，即可以放置棋子。现在有k个棋子，放置在棋盘区域中，并且要求任意两个棋子不能放在棋盘的同一行或者同一列。求放置k个棋子的方案数有多少。

如果要求不能放在同一行，那么在摆放的时候可以按行摆放。现在一行中找到能摆放的#的位置，然后放在这个位置上，dfs递归进入下一行，在下一行再找到合适的位置，如果剩余的棋子个数为0的时候，方案数目加一。

注意在选择位置的时候还要检查纵向是否已经摆放过棋子，若放过棋子，则此位置不能继续摆放棋子。

代码总览

#include <cstdio>
#include <cstring>
#include <algorithm>
#define nmax 10
using namespace std;
char mp[nmax][nmax];
bool visit[nmax][nmax];
int n,k,ans;
bool check(int x ,int y)
{
    if(mp[x][y] != '#' || visit[x][y] == true ) return false;
    else{
        for(int i = 0;i<n;++i){
            if(visit[i][y]) return false;
        }
    }
    return true;
}
void dfs(int line,int left)
{
    if(left == 0){
        ans++;
        return;
    }
    if( line>=n ) return;
    for(int i = 0;i<n;++i){
        if(check(line,i)){
            visit[line][i] = true;
            dfs(line+1,left-1);
            visit[line][i] =false;
        }
    }
    dfs(line+1,left);
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d %d",&n,&k) != EOF){
        memset(mp,0,sizeof(mp));
        memset(visit,0,sizeof(visit));
        ans = 0;
        if(n == -1 && k == -1) break;
        for(int i = 0;i<n;++i){
            scanf("%s",mp[i]);
        }
        dfs(0,k);
        printf("%d\n",ans);
    }
    return 0;
}