[kuangbin]专题一 简单搜索 棋盘问题 POJ - 1321【DFS】

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

样例解释：对于第一组数据，一个棋子可以放在任意#处，故有2种情况，第二组数据中，4个棋子只能固定放在4个#处，故只有一种情况。

题目链接：https://cn.vjudge.net/problem/POJ-1321

代码如下：

#include 
#include 
using namespace std;
static const int MAXN=8;
char mp[MAXN+2][MAXN+2];
bool vis[MAXN+2];//vis用来标记某列是否被选
int n,k,ans;
void dfs(int x,int cnt)//用x表示当前行数，cnt表示已放棋子数
{
    if(cnt==k)//出口，如果已放棋子数量达到k，则表示一种情况完成
    {
        ans++;
        return;
    }
    for(int i=x;i<n;i++)
        for(int j=0;j<n;j++)
            if(mp[i][j]=='#' && vis[j])
            {
                vis[j]=false;
                dfs(i+1,cnt+1);//继续往下一行，下一个棋子进行搜索
                vis[j]=true;//回溯
            }
}
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0),cout.tie(0);
    while(cin>>n>>k && n!=-1 && k!=-1)
    {
        memset(mp,0,sizeof(mp));
        memset(vis,true,sizeof(vis));
        ans=0;//多组数据，需要初始化
        for(int i=0;i<n;i++)
            cin>>mp[i];
        dfs(0,0);
        cout<<ans<<endl;
    }
    return 0;
}