poj1321——dfs回溯

POJ 1321  DFS回溯+递归枚举

棋盘问题

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 24813		Accepted: 12261

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;
}