POJ 1321--棋盘问题【DFS】

棋盘问题

Description

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

Input

输入含有多组测试数据。

每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n

当为-1 -1时表示输入结束。

随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。

Output

对于每一组数据，给出一行输出，输出摆放的方案数目C （数据保证C<2^31）。

Sample Input

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

Sample Output

2
1

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

int map[10][10];
int vis[10];// 标记哪一列用过了
int n, k, ans;

void dfs(int i, int num){//i 代表行数，从第一行开始便利，到最后一行
    if(num == k){
        ans++;
        return ;
    }
    if(i > n)
        return ;
    for(int j = 1; j <= n; ++j)//便利，每一列
        if(map[i][j] && !vis[j]){
            vis[j] = 1;
            dfs(i + 1, num + 1);
            vis[j] = 0;
        }
    dfs(i + 1, num);//可能某一行没有棋盘区域，所以不能放棋子，进入下一行
}

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

版权声明：本文为博主原创文章，未经博主允许不得转载。