当前位置: 代码迷 >> 综合 >> poj 1321 棋盘问题 题解
  详细解决方案

poj 1321 棋盘问题 题解

热度:67   发布时间:2024-01-14 21:04:33.0
棋盘问题
Time Limit: 1000MS   Memory Limit: 10000K
Total Submissions: 30740   Accepted: 15208

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


题目大意:POJ上少有的汉字题面,题目意思简单明了,读完题就想起了之前的八皇后问题,八皇后问题是八行八列放八个皇后,这个题是n行n列放k个棋子,同样考察的是深度搜索,这道题难点在回溯上,当放完K个棋子后要回溯把之前放的棋子重新拿起继续搜索!

上代码:

#include <iostream>
#include<string.h>
using namespace std;
int map[110][110],vis[110][110];
int n,k,sum,num;
int check(int x,int y)
{
    int i;
    if(x==0&&map[x][y]) return 1;
    if(!map[x][y]) return 0;
    for(i=0; i<x; i++)
    {
        if(vis[i][y])
            return 0;
    }
    for(i=0; i<y; i++)
    {
        if(vis[x][i])
            return 0;
    }
    return 1;
}
void dfs(int x)
{
    int i;
    if(num==k)
    {
        sum++;
        return;
    }
    if(x>=n) return;
    for(i=0; i<n; i++)
    {
        if(check(x,i))
        {
            vis[x][i]=1;
            num++;
            dfs(x+1);
            vis[x][i]=0;
            num--;
        }
    }
    dfs(x+1);
}
int main()
{
    int i,j;
    char c;
    while(cin>>n>>k&&n!=-1&&k!=-1)
    {
        sum=0;
        num=0;
        memset(map,0,sizeof(map));
        memset(vis,0,sizeof(vis));
        for(i=0; i<n; i++)
        {
            for(j=0; j<n; j++)
            {
                cin>>c;
                if(c=='#') map[i][j]=1;
            }
        }
        dfs(0);
        cout<<sum<<endl;
    }
// cout << "Hello world!" << endl;
    return 0;
}