POJ 1321 - 棋盘问题 ( 经典DFS )

题意

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放k个棋子的所有可行的摆放方案C。

思路

经典DFS
用一个all记录摆放种类, 一个now记录目前为止放了几个棋子, 当now == k时 all++, return;

AC代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#define mst(a) memset(a, 0, sizeof(a));
using namespace std;
const int maxn = 10;
int g[maxn][maxn];
int vis[maxn]; //记录某列是否已放过棋子
int n, k, all, now;

void init(){
    mst(g);
    all = 0;
    now = 0;
}

void dfs( int x ){
    if( now == k ){
        all++;
        return;
    }
    if( x >= n )  return;
    for( int j = 0; j < n; j++ ){
        if( vis[j] == 0 && g[x][j] == 1 ){
            vis[j] = 1;
            now++;
            dfs(x+1);
            vis[j] = 0;
            now--;
        }
    }
    dfs(x+1);
    return;
}

int main()
{
    while( scanf("%d%d",&n, &k) == 2 && n != -1 ){
        init();
        getchar();
        mst(g);
        char ch;
        for( int i = 0; i < n; i++ ){
            for( int j = 0; j < n; j++ ){
                scanf("%c",&ch);
                if( ch == '#' ) g[i][j] = 1;
                else if( ch == '.' ) g[i][j] = 0;
            }
            getchar();
        }
        dfs(0);
        printf("%d\n", all);
    }
    return 0;
}