[POJ1321]棋盘问题(DFS)

题目

在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放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

思路

这道题可以理解为深度优先遍历一棵解答树,解答树的深度是棋盘的行数,同深度的结点以父结点或列的不同区分,区分列的不同的方法是用一个数组column来记路该路径是否在这一列下过。解答树叶的个数就是最后的答案。

代码

#include <cstdio>
#include <cstring>

int n, K;
int ans;
char map[10][10];
int column[10];                 //记录这一列是否有棋子

void dfs(int r, int k) {
    if (r == n || !k) {         //r是行,当遍历完所有行时,若棋子都用了,则ans++
        if (k == 0)  ans++;
        return;
    }
    for(int i = 0; i < n; i++) {
        if (map[r][i] == '#' && !column[i]) {
            column[i] = 1;
            dfs(r+1, k-1);
            column[i] = 0;       //撤销 
        }
    }
    dfs(r+1, k);                 //这一行的遍历完,什么棋也不下就跳到下一行。      
    return;
}

int main() {
    //freopen("input.txt", "r", stdin);
    while(scanf("%d%d", &n, &K) != EOF && n != -1 && K != -1) {
        ans = 0;
        for(int i = 0; i < n; i++) {
            scanf("%s", map[i]);
        }
        memset(column, 0, sizeof(column));
        dfs(0, K);
        printf("%d\n", ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sunmaoxiang/article/details/80682965