POJ 1321 棋盘问题

Time Limit: 1000MS Memory Limit: 10000K

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

思路：

简单深搜题，具体我们来看下面两种方法的代码吧~

AC代码如下：

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
char str[10][10],board[10][10],visit[10];
int tot=0,k,n;
bool feasible(int x,int y)
{
    int i,j;
    if(str[x][y]=='#')
    {
        for(j=0;j<=y;j++)
            if(board[x][j]==1)
                return false;
        for(i=0;i<=x;i++)
            if(board[i][y]==1)
                return false;
        return true;
    }
    return false;
}
void dfs(int cur,int chess)
{
    if(chess==k)
    {
        tot++;return ;
    }
    else
    {
        while(cur<n)
        {
            for(int j=0;j<n;j++)
            {
                if(feasible(cur,j))
                {
                    board[cur][j]=1;
                    dfs(cur+1,chess+1);
                    board[cur][j]=0;
                }
            }
            cur++;
        }
    }
}

int main()
{
    int i,j;
    while(~scanf("%d%d",&n,&k))
    {
        tot=0;
        memset(board,0,sizeof(board));
        if(n==-1&&k==-1)
            break;
        for(i=0;i<n;i++){
            getchar();
            for(j=0;j<n;j++)
                scanf("%c",&str[i][j]);
        }
        dfs(0,0);
        printf("%d\n",tot);
    }
    return 0;
}

AC代码（方法二）：

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
char str[10][10],visit[10];
int tot=0,k,n;

void dfs(int cur,int chess)
{
    if(chess==k)//当所有的棋子都放在棋盘上时
    {
        tot++;return ;
    }
    else
    {
        while(cur<n)//从第一行依次遍历
        {
            for(int j=0;j<n;j++)
            {
                if(str[cur][j]=='#'&&!visit[j])//如果符合要求，那么这一列肯定没有棋子，所以我们只需用一个一维的数组来标记当前列有无棋子，而对于行的话，每次都是取下一行，因此不会出现重复情况。
                {
                visit[j]=true;
                    dfs(cur+1,chess+1);
                    visit[j]=false;
                }
            }
            cur++;
        }
    }
}

int main()
{
    int i,j;
    while(~scanf("%d%d",&n,&k))
    {
        tot=0;
        if(n==-1&&k==-1)
            break;
        for(i=0;i<n;i++){
            getchar();
            for(j=0;j<n;j++)
                scanf("%c",&str[i][j]);
        }
        dfs(0,0);
        printf("%d\n",tot);
    }
    return 0;
}