HDU 1321 棋盘问题

最新推荐文章于 2021-02-07 19:20:47 发布

原创最新推荐文章于 2021-02-07 19:20:47 发布 · 698 阅读

1 ·

CC 4.0 BY-SA版权

dfs 专栏收录该内容

8 篇文章

订阅专栏

本文探讨了一个特定的棋盘摆放问题，即在一个不规则形状的棋盘上摆放多个棋子，要求任意两个棋子不能在同一行或同一列。通过递归深度优先搜索算法求解所有可行的摆放方案数量。

棋盘问题

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 47741		Accepted: 23097

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

今天用一个全局变量count，快一个小时没编译过去，后来才发现count时C++的algorithm头文件一个库函数，怪不得死都不给过woc~╮(╯▽╰)╭菜鸡就得买教训

#include <iostream>
#include <string>
#include <stack>
#include <queue>
#include <map>
#include <algorithm>

using namespace std;

char vis[8][8];
int isSafe[8];
int sum;
int n;
int k;


bool Safe(int x,int y) {
    if(isSafe[x] && vis[x][y] == '#') {
        return 1;
    }
    return 0;
}

void dfs(int y,int num) { //mum已经放入棋盘的棋子个数
    if(num == k) { //全部放完为一种情况
        sum++;  
        return;
    }
    if(y >= n) {//列结束
        return;
    }
    for(int i = 0;i < n;i++) {
        if(Safe(i,y)) {
            isSafe[i] = 0;//第i行被放过
            dfs(y+1,num+1);//去下一列放置
            isSafe[i] = 1;//相当于把第i行放置的棋子给拿掉
        }
    }
    dfs(y+1,num);//当每列结束后去进行下一行
}

int main(void) {
    while(cin >> n >> k) {
        if(n == -1 && k == -1) {
            break;
        }

        sum = 0;
        for(int i = 0;i < n;i++) {
            for(int j = 0;j < n;j++) {
                cin >> vis[i][j];
            }
        }
        for(int i = 0;i < n;i++) {
            isSafe[i] = 1;
        }
        dfs(0,0);
        cout << sum << endl;
    }
    
    return 0;
}