(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

分析与解答

k<=n,我觉得这个和八皇后不同一点就是八皇后列从第一行一直找到最后第n行,但是这个可以在某一行不放棋子。于是我们dfs里面就要考虑写一个二重循环,行和列都要考虑到,我们假设固定行,就是说,每次行从第x行开始,就算回溯了,还可以从第x+i行继续调用,从而考虑到所有情况
由于是一行一行的调用,因此已经放置的棋子必然不再同一行,那现在就需要考虑怎么整一个条件能够去判断是不是在同一列,额当然了,用一个标记数组visit,如果如果标志数组visit[j]=0说明第j列没有用过,那我们就可以大胆的放到第j列了
dfs(x,y)的意思是在第x行准备放第y+1个棋子,额就是现在已经放了y个棋子了,我们固定了这第x行,从第一列开始找,我们放棋子的条件是!visit[j]&&mp[i][j]==’#’,找到一满足条件的列,我们就把棋子放这一列上,放完这一行的棋子,我们就继续递归找下一行,递归完了还要清除标志数组,因为除了在第x行的第j列放棋子,仍有可能在其他列仍然能放棋子。而dfs的返回条件就是现在放的棋子个数y已经等于k了,那就return了

代码参考:
https://www.cnblogs.com/ECJTUACM-873284962/p/6747947.html

#include<iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
using namespace std;
int visit[20];
char mp[20][20];
int ans;
int k;
int n; 
int DFS(int x,int y)
{
    if(y==k)
    {
        ans++;
        return 0;
    }
    for(int i=x;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(!visit[j]&&mp[i][j]=='#')
            {
                visit[j]=true;
                DFS(i+1,y+1);
                visit[j]=false;
            }
        }
    }
    return 0;
}
int main()
{
    while(cin>>n>>k)
    {
        if(n==-1&&k==-1)
            break;
        memset(visit,false,sizeof(visit));
        memset(mp,false,sizeof(mp));
        for(int i=0;i<n;i++)
            cin>>mp[i];
        ans=0;
        DFS(0,0);
        cout<<ans<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40828914/article/details/81459237