【POJ 1321】 棋盘问题 DFS 详解

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

题意：如题

思路（简单DFS+特殊情况考虑）：

纯DFS题，类似于八皇后，这里只需要行列不冲突即可。方法就是用vis数组标记列，然后dfs每一层，对于每一层的每个位置，若!vis[i]&&grid[pos][i]（grid[pos][i]为1代表第pos行，i列是’# '）成立，就在这个点开始往下继续搜索dfs（pos+1），同时在这之前计数++，vis标记当前列位置。然后递归出口就是找到k个位置放，或者到边界了。但是要注意的一种情况就是比如
11
11（1代表#）
k=1的时候这种情况，第一行每次搜完就因为计数==k而return了，不会继续搜第二行，这时候就需要每行搜索完后强制搜索下一层，把所有情况都枚举一遍。

AC代码：

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define maxn 100000+500
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
const int maxm = 100000+5;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll x = 0;char ch = getchar();while(ch>'9'||ch<'0') ch = getchar();while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0',  ch = getchar();return x; }
int n,k;
int grid[10][10];
int vis[10];
int ans = 0;
int cur = 0;
void init()
{
    memset(vis,0,sizeof(vis));
    ans = 0;
}

void dfs(int pos)
{
    if(cur==k)
    {
        ans ++;
        return ;
    } if(pos==n+1) return ;
    for(int i=1;i<=n;i++)
    {
        if(!vis[i]&&grid[pos][i])
        {
            cur++;
            vis[i] = 1;
            dfs(pos+1);
            vis[i] = 0;
            cur --;
        }
    }
    dfs(pos+1);
}

int main()
{
    while(cin>>n>>k&&n!=-1)
    {
        init();
        char c = getchar();
        string s;
        for(int i=1;i<=n;i++)
        {
            getline(cin,s);
            for(int j=1;j<=n;j++)
            if(s[j-1]=='#') grid[i][j] = 1; else grid[i][j] = 0;
        }
        dfs(1);
        cout<<ans<<endl;
    }
    return 0;
}