棋盘问题 POJ1321 DFS kuangbin带我飞

来源:互联网 发布:js正则表匹配标点符号 编辑:程序博客网 时间:2024/06/05 15:32

A - 棋盘问题
Time Limit:1000MS Memory Limit:10000KB 64bit IO Format:%I64d & %I64u
Submit

Status

Practice

POJ 1321
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

典型的八皇后问题,只不过这回不是一个规则的棋盘,只有带“#”的位置才是有效的棋盘区域,不过换汤不换药,只要添加额外的递归来跳过所有的无效区域就可以了,所以,算法可以这样描述:

1.从(0,0)点开始,从左向右扫描
2.当遇到“#”,将这个位置设置为“.”,然后将这个位置所在的行和列都标记为已经访问,也就是说今后再寻找的时候不能再找这一行和这一列了,然后直接从下一行的第一个开始搜索,回溯的时候需要恢复状态
3.不管这个位置是什么,当下一次扫描这个位置的时候,都要从这个位置的下一个开始扫描
4.当用完所有的棋子以后,表示找到了一种方案,将方案数增一
5.输出所有的方案数

#include <iostream>#include <cmath>#include <stdio.h>#include <string>#include <cstring>#include <map>#include <set>#include <vector>#include <stack>#include <queue>#include <iomanip>#include <algorithm>#include <memory.h>using namespace std;char Map[20][20];int n,k;bool visitr[20];//保存行的访问状态bool visitc[20];//保存列的访问状态long long cas;void init(){    memset(visitr,0,sizeof(visitr));    memset(visitc,0,sizeof(visitc));    cas=0;}void input(int n){    int i,j;    getchar();//使用gets输入的时候,如果在之前还有读入的数据,需要用getchar吸收一个回车    for(i=0;i<n;i++)    {        gets(Map[i]);    }}void dfs(int r,int c,int count){    if(count==k)//所有的棋子都已经用完    {        cas++;        return;    }    if(r==n)//扫描到最后一行        return;    if(visitr[r]!=1&&visitc[c]!=1&&Map[r][c]=='#')    {        Map[r][c]='.';        visitr[r]=1;        visitc[c]=1;        dfs(r+1,0,count+1);//直接扫描下一行        Map[r][c]='#';//回溯        visitr[r]=0;        visitc[c]=0;    }        if(c==n)//扫描下一个位置            dfs(r+1,0,count);        else            dfs(r,c+1,count);}int main(){    while(cin>>n>>k)    {        if(n==-1&&k==-1)            break;        init();        input(n);        dfs(0,0,0);        cout<<cas<<endl;    }    return 0;}
0 0
原创粉丝点击