[kuangbin带你飞]专题一 简单搜索

题目描述

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

解题思路

这个问题其实是八皇后问题的一个变型题。不同的的地方有三点：1、这里的棋子数量不一定和棋盘的长宽相等。2、棋盘不是所有地方都可以落子。3、不需要考虑对角的问题。
和八皇后的问题一样，我们这里还是使用搜索的算法。先选行放棋子，然后再选列放棋子，当棋子放完，就代表找到一种解法。八皇后问题可以通过判断是否所有行都占满来确定是否找到解，而这道题由于棋子可能会小于行数，因此只能通过判断棋子是否放完来找解。这种搜索的方法类似于DFS的思想。行方向是优先的，当行方向不能行的通或者已经找到一个解，就开始回溯，再列方向上再继续依次找解。下面给出ac代码

下面展示一些 内联代码片。

#include<iostream>
#include<stdio.h>
using namespace std;

char a[10][10];
int clo[10] = {-1,-1,-1,-1,-1,-1,-1,-1,-1,-1};
int num = 0;
int n,k;
int m = 0;
void dfs(int x)
{
    if(m==k)
    {
        num = num+1;
        return;
    }
    if(x>n-1) return;
    for(int i  = 0;i<n;i++)
    {
        if(clo[i]==-1&&a[x][i]=='#')
        {
            clo[i] = x;
            m++;
            dfs(x+1);
            clo[i] = -1;
            m--;
        }
    }
    dfs(x+1);
}
int main()
{

    while(cin>>n>>k&&n!=-1&k!=-1){
        for(int i = 0;i<n;i++){
            cin>>a[i];
        }
        dfs(0);
        cout<<num<<endl;
        num = 0;
        m = 0;
        for(int i = 0;i<10;i++) clo[i] = -1; 
    }

//    clo = {-1,-1,-1,-1,-1,-1,-1,-1,-1,-1};

 }