POJ-1321 棋盘问题

传送门：POJ - 1321

题目描述

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

输入描述

输入含有多组测试数据。

每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n

当为-1 -1时表示输入结束。

随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。

输出描述

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

题意分析

搜

样例输入

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

样例输出

2
1

AC代码

#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <iomanip>
#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef map<int, int> mii;
typedef map<ll, ll> mll;
#define ull unsigned long long
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int maxn=1e6+10;
const ll mod=1e9+7;
const int INF = 0x3f3f3f3f;
char a[10][10];
int vis[20];
int n,k,sum;

void dfs(int d,int ans) {

    if(ans==k) {
        sum++;
        return ;
    }
    if(d==n)return ;
    for(int i=0; i<n; i++) {
        if(!vis[i]&&a[d][i]=='#') {
            vis[i]=1;
            dfs(d+1,ans+1);
            vis[i]=0;
        }
    }
    dfs(d+1,ans);
}

int main() {
    IOS;
    while(cin>>n>>k) {
        memset(a,0,sizeof(a));
        memset(vis,0,sizeof(vis));
        if(n==-1)break;
        int i,j;
        sum=0;
        for(i=0; i<n; i++) {
            for(j=0; j<n; j++) {
                cin>>a[i][j];
            }
        }
        dfs(0,0);
        cout<<sum<<endl;
    }
}