poj1321——棋盘问题(DFS)

来源:互联网 发布:金山软件有哪些 编辑:程序博客网 时间:2024/06/01 10:30

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

考虑每遇到一个#就放棋子,然后深搜在这个#上放棋子的可能性。简单来说就是遇到每个#就遍历一遍棋盘放置,然后回溯。数据规模不大,所以可以跑完

#include <iostream>#include <algorithm>#include <cstring>#include <cmath>#include <cstdio>#include <set>#include <vector>#include <iomanip>#include <stack>#include <map>#include <queue>#define MAXN 1010#define mod 2012#define INF 0x3f3f3f3fusing namespace std;char m[10][10];int k,vis[10][10],ans,n;bool check(int x,int y){    for(int j=0;j<n;++j)        if(vis[x][j])        return false;    for(int i=0;i<n;++i)        if(vis[i][y])        return false;    return true;}void dfs(int x,int y,int s){    if(s==k)    {        ans++;        return;    }    vis[x][y]=1;    for(int i=x+1; i<n; ++i)        for(int j=0; j<n; ++j)        {            if(m[i][j]=='#'&&!vis[i][j]&&check(i,j))            {                vis[i][j]=1;                dfs(i,j,s+1);                vis[i][j]=0;            }        }    vis[x][y]=0;}int main(){    ios::sync_with_stdio(false);    while(cin>>n>>k)    {        if(n==-1&&k==-1)            break;        for(int i=0; i<n; ++i)            for(int j=0; j<n; ++j)                cin>>m[i][j];        ans=0;        memset(vis,0,sizeof(vis));        for(int i=0; i<n; ++i)            for(int j=0; j<n; ++j)                if(m[i][j]=='#')                    dfs(i,j,1);        cout<<ans<<endl;    }    return 0;}
0 0
原创粉丝点击