poj1321棋盘问题

poj1321 棋盘问题

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的基本应用，因为每行和每列是不能有两颗棋子的，所以标记第几列是否有棋子就可以了。然后搜索每一行满足条件的。注意满足条件的位置是可以不放棋子的，所以要考虑当前位置不放子的情况。

#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;
int n,k;
int maze[10][10];
int col[10];


long long out = 0;
int k_num = 0;

int dfs(int num,int k_num)
{
    if(k_num == k)
    {
        out++;return 0;
    }
    if(num == n)
    {return 0;}


    for(int i = 0;i < n; i++)
        if(!col[i] && !maze[num][i])
        {
            col[i] = 1;
            dfs(num+1,k_num+1);
            col[i] = 0;
        }
    dfs(num+1,k_num);//当前位置不放子
}

int main()
{
    while(scanf("%d %d",&n,&k) != EOF && n != -1)
    {
        for(int i = 0; i < n; i++)
        {
            char data[15];
            scanf("%s",data);

            for(int j = 0; j < n ; j++)
            {
                if(data[j] == '#') maze[i][j] = 0;
                else maze[i][j] = 1;
            }
        }

        out = 0;k_num = 0;
        memset(col,0,sizeof(col));
        dfs(0,0);

        printf("%I64d\n",out);
    }
    return 0;
}

真的是感觉处理成0和1比读字符方便多了，不需要考虑回车怎么办。
还有这个的数据2^31肯定是没int大的，虽然我也没有试改成int会不会出问题。