棋盘问题(回溯/DFS) POJ1321

在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放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皇后问题很像,n皇后还得判断对角线,这道题不需要.建议看一下n皇后问题,我看完后就会写了,一个套路,回溯算法.

这题还有一个和n皇后后问题的区别就是,n皇后要每行都得放上一个棋子,但是这道题不一定每行都要放上棋子.所以有了如下操作

在一次循环后,在剩余棋子不变的情况下,,跳到下一行去,那么原来这行就是空的了.

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int N=10;
char s[N][N];
int a[N];
int n,k;
int ans;
void dfs(int now,int end,int step){
    for(int i=1;i<=n;i++){
        if(a[i]==0&&s[now][i]=='#'){
            if(step==1) ans++;
            else if(now!=end) a[i]=1,dfs(now+1,end,step-1),a[i]=0;//回溯
        }
    }
    if(now!=end) dfs(now+1,end,step);
}
int main()
{
    while(cin>>n>>k&&n!=-1){
        memset(a,0,sizeof(a));
        ans=0;
        for(int i=1;i<=n;i++) scanf("%s",s[i]+1);
        dfs(1,n,k);
        cout<<ans<<endl;
    }
    //cout << "Hello world!" << endl;
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/-yjun/p/10646435.html