博客

poj 1321 棋盘问题

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放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

类似八皇后的题目，DFS入门级别

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
typedef long long ll;
const int N = 10;
char a[N][N];
bool row[10];//标记每列是否有棋子
int n,k,cot=0,sum=0;
void dfs(int x) //对第x行搜索
{
    if(k==cot){
        sum++;
        return;
    }//结束条件，已经放入要求个数
    if(x>n)return;//超出边界
    for(int j=1; j<=n; j++)
        if(a[x][j]=='#'&&row[j]==0){ //满足条件即可摆棋子
            cot++;
            row[j]=1;
            dfs(x+1);//搜索下一行
            row[j]=0;//搜索之后回溯剪枝，去除标记
            cot--;
        }
    dfs(x+1);//直接搜索下一行
}
int main()
{
    while(cin>>n>>k&&n!=-1&&k!=-1){
        sum=cot=0;
        memset(row,0,10);//初始化
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++){
                cin>>a[i][j];
            }
        dfs(1);//从第一行开始搜索
        printf("%d\n",sum);
    }
    return 0;
}

算法标签:DFS 回溯