码迷,mamicode.com
首页 > 其他好文 > 详细

poj1321——dfs回溯

时间:2015-03-13 00:17:13      阅读:166      评论:0      收藏:0      [点我收藏+]

标签:

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;
}
poj1321_dfs回溯

 

poj1321——dfs回溯

标签:

原文地址:http://www.cnblogs.com/--560/p/4334086.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!