POJ1321棋盘问题【搜索】

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

输入含有多组测试数据。 
每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n 
当为-1 -1时表示输入结束。 
随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。 

 

分析：直接暴力搜索就可以了

 

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 9;
int vis[maxn][maxn];
char mat[maxn][maxn];
int n, k;
int vis2[maxn];
int ans;

void dfs(int i, int num) {
//    printf("%d %d\n", i, num);
    if(num == k) {
        ans ++;
        return;
    } 
    if(i > n) return ;
    for(int j = 1; j <= n; j++) {
        if(vis[i][j] == 0 && !vis2[j]) {
            vis2[j] = 1;
            dfs(i + 1, num + 1);
            vis2[j] = 0;
        }
    }
    dfs(i + 1, num);
}

int main() {
    while(scanf("%d %d",&n, &k) ) {
        if(n == -1 && k == -1) break;
        for(int i = 0; i < n; i++) scanf("\n%s", mat[i]);
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                if(mat[i][j] == '#') {
                    vis[i + 1][j + 1] = 0;
                } else vis[i + 1][j + 1] = 1;
            }
        }
        ans = 0;
        memset(vis2, 0, sizeof(vis2));
        dfs(1, 0);
        printf("%d\n", ans);
    }
    return 0;
}