简单搜索-POJ1321棋盘问题(n皇后问题升级版)

先说n皇后问题
在一个n*n的棋盘上放置n个皇后
使得这n个皇后不在同一横行同一横列和对角线上
n皇后问题的回溯法

POJ1321棋盘问题

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

题意

在n*n的棋盘上放置k个棋子,问共有多少种放法

思路

对每一行dfs,到第x行对每一列遍历,看这一列是否被占用,不被占用就可以放置在这里,也可以选择不放,因为空比棋子多,,每当放够了k个棋子ans就++,就酱

AC代码

#include<cstdio>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

#define PI         acos(-1)
#define inf        0x3f3f3f3f
#define EPS        1e-6
#define mem(a, b)  memset(a, b, sizeof(a))
#define ll long long
#define mian main

int n, k;
bool vis[10];
char mp[10][10];
int ans, num;
void dfs(int x)
{

    if(num == k)
    {
         ans ++;
         return;
    }
    if(x >= n)
        return ;
    for(int i = 0; i < n; i++)
    {
        if(!vis[i] && mp[x][i] == '#')
        {
            vis[i] = 1;
            num ++;//放置在此符合条件的位置
            dfs(x+1);
            vis[i] = 0;//也可以不放在这里,为下一行做准备
            num--;
        }
    }
    dfs(x+1);//如果这一行的每一列都不满足条件,直接遍历下一行

}
int main()
{
    while(cin>>n>>k && n != -1 && k != -1)
    {
        mem(vis, 0);
        ans = 0;
        num = 0;
        for(int i = 0; i < n; i++)
        {
            for(int j = 0; j < n; j++)
            {
                cin>>mp[i][j];
            }
        }
        dfs(0);//从第0行开始
        cout<<ans<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/wuswi0412/article/details/81209948