棋盘问题 POJ1321 DFS kuangbin带我飞

A - 棋盘问题
Time Limit:1000MS Memory Limit:10000KB 64bit IO Format:%I64d & %I64u
Submit

Status

Practice

POJ 1321
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

典型的八皇后问题，只不过这回不是一个规则的棋盘，只有带“#”的位置才是有效的棋盘区域，不过换汤不换药，只要添加额外的递归来跳过所有的无效区域就可以了，所以，算法可以这样描述：

1.从（0，0）点开始，从左向右扫描
2.当遇到“#”，将这个位置设置为“.”，然后将这个位置所在的行和列都标记为已经访问，也就是说今后再寻找的时候不能再找这一行和这一列了，然后直接从下一行的第一个开始搜索，回溯的时候需要恢复状态
3.不管这个位置是什么，当下一次扫描这个位置的时候，都要从这个位置的下一个开始扫描
4.当用完所有的棋子以后，表示找到了一种方案，将方案数增一
5.输出所有的方案数

#include <iostream>
#include <cmath>
#include <stdio.h>
#include <string>
#include <cstring>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <iomanip>
#include <algorithm>
#include <memory.h>
using namespace std;
char Map[20][20];
int n,k;
bool visitr[20];//保存行的访问状态
bool visitc[20];//保存列的访问状态
long long cas;
void init()
{
    memset(visitr,0,sizeof(visitr));
    memset(visitc,0,sizeof(visitc));
    cas=0;
}

void input(int n)
{
    int i,j;
    getchar();//使用gets输入的时候，如果在之前还有读入的数据，需要用getchar吸收一个回车
    for(i=0;i<n;i++)
    {
        gets(Map[i]);
    }
}


void dfs(int r,int c,int count)
{
    if(count==k)//所有的棋子都已经用完
    {
        cas++;
        return;
    }
    if(r==n)//扫描到最后一行
        return;
    if(visitr[r]!=1&&visitc[c]!=1&&Map[r][c]=='#')
    {
        Map[r][c]='.';
        visitr[r]=1;
        visitc[c]=1;
        dfs(r+1,0,count+1);//直接扫描下一行
        Map[r][c]='#';//回溯
        visitr[r]=0;
        visitc[c]=0;
    }

        if(c==n)//扫描下一个位置
            dfs(r+1,0,count);
        else
            dfs(r,c+1,count);



}



int main()
{
    while(cin>>n>>k)
    {
        if(n==-1&&k==-1)
            break;
        init();
        input(n);
        dfs(0,0,0);
        cout<<cas<<endl;
    }
    return 0;
}