POJ1321 简单搜索

题目:
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。

Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。

Output
对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。

找对做的方法思路就挺简单的。
思路:
先画几个图模拟一下,其实大概就是k个组合。不用管k多少,直接往下搜索,每次搜索完就缩减一行一列继续搜索,用一个数存以放棋盘个数等于k则另一个存多少种方法的数+1。用一维数组visit看这一列是否有确定过的棋盘。

#include"iostream"
#include"cstdio"
#include"stdlib.h"
#include"cmath"
#include"cstring"
#include"cstdlib"
#include"vector"
#include"stack"
#include"queue"
#include"set"
#include"map"
#include"algorithm"
#include <utility>
#include <iomanip>
#include <time.h>
#include<list>
const double PI=acos(-1.0);
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x&-x
#define inf 0x3f3f3f3f
#define pr pair<int,int>
//priority_queue<int,vector<int>,greater<int>> pp;
const int mod = 998244353 ;
const int M = 2e6 + 10;
const int limt = 1<<20;
const int INF = 1e9;
typedef long long ll;
int t,n,b,d,r,ox,oy,l,i,j;
int cnt,sum;
pair<int,int>num[10];
char mp[10][10];
int visit[10];
int k;
void dfs(int x)
{
   
    if(sum==k)
    {
   
        cnt++;
        return ;
    }
    if(x>n)
        return ;
    for(int i=1;i<=n;i++)
    {
   
        if(!visit[i]&&mp[x][i]=='#')
        {
   
            visit[i]=1;
            sum++;
            dfs(x+1);
            sum--;
            visit[i]=0;
        }
    }
    dfs(x+1);
}
int main()
{
   
    while(cin>>n>>k)
    {
   
        t=0;
        if(n==-1&&k==-1)
            break;
        for(int i=1;i<=n;i++)
        {
   
            for(int j=1;j<=n;j++)
            {
   
                cin>>mp[i][j];
            }
        }
        memset(visit,0,sizeof(visit));
        cnt=0;
        sum=0;
        dfs(1);
        cout<<cnt<<endl;
    }
    return 0;
}


全部评论

相关推荐

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