POJ - 1321 棋盘问题 【dfs】

题目简述

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放k个棋子的所有可行的摆放方案C。

数据范围

$n <= 8,k <= n$

简答

类似于八皇后问题，直接爆搜即可，不用剪枝。

点击查看代码

#include<iostream>
#include<cstdio>
using namespace std;
#define N 10
int ip(){
    int x=0,w=0;char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    return w?-x:x;
}
int n,k;
char a[N][N];
bool X[N],Y[N];
int ans;
void dfs(int x,int y,int s){
    if(y==n) y=0,x++;
    if(x==n){
        if(s==k) ans++;
        return;
    } 
    if(!X[x]&&!Y[y]&&a[x][y]=='#'){
        X[x]=1;Y[y]=1;
        dfs(x,y+1,s+1);
        X[x]=0;Y[y]=0;
    }
    dfs(x,y+1,s);
}
int main(){
    n=ip(),k=ip();
    while(n!=-1&&k!=-1){
        ans=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                cin>>a[i][j];
        dfs(0,0,0);
        printf("%d\n",ans);
        n=ip(),k=ip();
    }
    return 0;
}