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

简单dfs。要保证不在同一行和同一列。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <queue>
using namespace std;

char ma[10][10];int vis[10][10];
int sum;int n,k;

int dx[]={1,0,-1,0};
int dy[]={-1,1,0,0};

void do1(int i)
{
    for(int j=0;j<n;j++)
        vis[j][i]=1;
}

void do2(int i)
{
    for(int j=0;j<n;j++)
        vis[j][i]=0;
}

void dfs(int deep,int num)
{

    if(num==k+1)
    {
        sum++;
        return ;
    }

    if(deep>=n)return ;

    for(int i=0;i<n;i++)
    {
        if(!vis[deep][i]&&ma[deep][i]=='#')
        {
            do1(i);//将第i列所有的都标记
            dfs(deep+1,num+1);
            do2(i);//再标记回来
        }
    }
    dfs(deep+1,num);
}

int main()
{
    while(cin>>n>>k,n!=-1)
    {
        sum=0;
        for(int i=0;i<n;i++)
            cin>>ma[i];
        dfs(0,1);
        cout<<sum<<endl;
    }
}