棋盘问题

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

首先介绍一波DFS算法

DFS(Depth-First-Search)深度优先搜索算法,是搜索算法的一种。

https://blog.csdn.net/ldx19980108/article/details/76324307

然后复制来的分析

看到这个题目,第一反应就是dfs遍历求解,事实证明这是正确的。
那么接下来,问题就变成了解决以下几个难点:
1. 如何保证不重行,不重列。
2. 当k < n的时候,如何从这n行数据棋盘数据中,首先挑出有棋盘区域的行,再挑出选用其中的那几行。

首先,当k=n的时候,那么必然每一行都会用到一个数据,也就是说除非棋盘每一行都有棋盘区域,否则不可能有摆放位置。当k < n时候,需要从n行棋盘中,首先挑出有棋盘区域的m行,再从m行中选出k行来放棋子。这里会涉及一个组合的生成C(m, k)。当解决了组合数的问题后,就是保证不重行不重列的问题。初步设想是给棋盘位置设置一个vis数组,用来标记,然后通过:

bool conflict(int x, int y)
{
    for(int i=0; i<n; i++)
    {
        if(vis[x][i]==1)
            return true;
    }
    for(int i=0; i<n; i++)
    {
        if(vis[i][y]==1)
            return true;
    }
    return false;
}

来返回是否冲突。
看起来似乎一切顺利……
——————————————————————————分割线
然而!
最重要的问题,从哪儿开始dfs???!

问题源于上面的组合,当k < n 的时候,有些行是不使用的,那dfs可能就要通过循环控制起点,但是,这势必会造成极差的时间复杂度,而且一定会进行大量的重复子问题计算。

所以,上述方法不成立。

换思路。
——————————————————————————分割线

既然判断从哪儿dfs开始很困难,换句话说,也就是选择dfs(x, y)的参数a和b很困难。这说明,这道题dfs的参数根本就不应该是坐标!!!

那是什么?
我们在这里采用分治的思想,把问题一步一步拆分成子问题。
要解决k个棋子的问题,也就是要解决k-1个棋子的子问题再乘以第k个棋子的方法数;要解决k-1个棋子的问题,也就是要解决k-2个棋子的子问题…………如此递归下去,直至要解决0个棋子的问题的时候,就是到终点要return的时候了。

那么,我们换一种方法控制不充行,不重列。

首先,dfs的经典三层结构是亘古不变的真理:

dfs()
{
    if条件判断——(到终点) 
    {
        操作……
        return;
    }
    if……省略一些条件判断(剪枝优化,比如一些没必要的continue或者计算过的子问题的标记备忘录直接存取)
    for()
        if……满足条件的筛选
        修改标记;
        dfs();
        改回标记;
}

(注:有些情况下是不需要dfs三层结构第三层:改回标记 这一操作的,见bnuoj12921或bnuoj1681, 这种情况多是因为:
需要遍历全部满足题意的点,是纯粹的深度优先搜索遍历全部可走点)

从这个角度出发,我们思考一下:
首先,return的条件是什么?肯定是到终点了,所以肯定需要进行一些操作,比如某个count++,然后return
if条件判断,剪枝优化,毕竟dfs……毕竟递归……能优化尽量优化……
接下来是dfs递归的核心部分:
我们不妨从第0行开始,这时候需要处理的是k个棋子,对第0行遍历,找到一个满足题意的‘#’区域后,我们知道以后这一列都不能用了,我们vis数组可以只建立为一维数组,只存储用过的行就可以,然后深层搜索,即dfs,这时候,我们就要从第1行开始搜索了(正因为这样,vis才不用存储行是否用过的冲突问题),此时需要处理的是k-1个棋子,我们同样是对第1行所有元素都遍历,找‘#’,只不过对于之前vis标记过得那一列,我们也不选用,找到后,再次标记,然后dfs……………………dfs进入第i行,对第i行遍历,当前需要处理的是k-i个棋子的问题,筛掉非棋盘区域和之前标记过得列,找到’#’,标记,然后dfs()………………知道某一次dfs进入后,发现当前处理棋子数量为0,说明到头了,这是找到了一种方法 ,count++,return,假设这是第一次return,那么将返回到第k-1行,继续寻找满足的‘#’,重复…………知道某一次return会到第0行了,那么,将从第1行继续作为dfs的第一次起始,这就相当于改变控制了dfs起始的位置。

 

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, k, ans;
char str[10][10];
int vis[100];

void dfs(int r, int k)
{
    if(k==0)
    {
        ans++;
        return;
    }

    for(int i=r; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            if(str[i][j]=='.' || vis[j]==1) continue;
            vis[j] = 1;
            dfs(i+1, k-1);
            vis[j] = 0;
        }
    }
}

int main()
{
    //int n,k;
    //char str[10][10];
    while (scanf("%d %d",&n,&k)!=EOF){
        if(n==-1&&k==-1)break;
        memset(str, '\0', sizeof(str));
        memset(vis, 0, sizeof(vis));
        ans=0;
        for(int i=0;i<n;i++){
            scanf("%s",str[i]);
        }

        dfs(0, k);
        printf("%d\n", ans);
    }
    //cout << "Hello world!" << endl;
    return 0;
}

 

全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务