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

思路：用N皇后问题的方法按照行顺序进行dfs回溯，用int型vis数组记录可行域，增量和减量控制可行域的改变；由于k可以小于n，所以需要递归枚举；注意边界控制，由于边界控制不当WA了好多次。。

用了几个小时终于AC了

//poj1321——dfs回溯
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>

using namespace std;

const int maxn=30;
const int INF=(1<<28);

int n,k;
int map[maxn][maxn];
int cnt;

void put(int x,int y)
{
    for(int i=1;i<=n;i++){
        map[i][y]++;
        map[x][i]++;
    }
}

void remov(int x,int y)
{
    for(int i=1;i<=n;i++){
        map[i][y]--;
        map[x][i]--;
    }
}

void search(int cur,int num)
{
    for(int i=1;i<=n;i++){
        if(map[cur][i]) continue;
        bool flag=0;
        if(num==k){
            cnt++;
            flag=1; //边界控制关键处在这里。不能直接return，因为下面要将放置的棋子移除
        }
        put(cur,i);  //放置棋子
        if(!flag){
            for(int j=cur+1;j<=n;j++){
                search(j,num+1);
            }
        }
        remov(cur,i); //移除棋子
    }
}

int main()
{
    while(cin>>n>>k,n!=-1,k!=-1){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                char ch;cin>>ch;
                if(ch==‘#‘) map[i][j]=0;
                else map[i][j]=1;
            }
        }
        cnt=0;
        for(int i=1;i<=n;i++) search(i,1); //递归枚举
        cout<<cnt<<endl;
    }
    return 0;
}

poj1321_dfs回溯