题目描述:n为正方形棋盘一维长度,k为需放置的棋子数,在放置棋子时,任意两个棋子必须不同行不同列。给出n,k以及棋盘形状,求放置k个棋子的方案数。
思路:有点像八皇后问题,需要dfs解决。dfs的重点是如何简化搜索,防止重复计数。一开始没有考虑复杂度,重复搜索方案,最后结果除以对应组合数,会超时而且可能爆int。
为了保证一种情况只被搜到一次,我们需要确定搜索的合适的序,达到在多维空间里穷尽所有情况不重不漏。
1.所有棋盘上合法的点都依次作为第一个下棋的点
2.每次搜索时保证有序,只会搜索当前棋子行的后面的行。
这样可以保证枚举所有的情况,并且所有的情况都不相同。
AC代码
//#include<bits/stdc++.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
typedef long long ll;
typedef long double db;
typedef pair<int,int> pii;
#define de(x) cout << #x << "=" << x << endl
#define rep(i,a,b) for(int i=a;i<(b);++i)
#define all(x) (x).begin(),(x).end()
#define sz(x) (int)(x).size()
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define lb(x) (x&-(x))
const int N = 101010;
int n,k;
int cnt;
char grid[10][10];
int visr[10];
int visc[10];
int chk(int ii,int jj)
{
if(visr[ii]==0&&visc[jj]==0&&ii>0&&jj>0&&ii<=n&&jj<=n&&grid[ii][jj]!='.') return 1;
else return 0;
}
//思路 通过搜索确定
void dfs(int x,int y,int depth)
{
// printf("now pos is (%d,%d) depth is %d\n",x,y,depth);
visr[x]=visc[y]=1;
if(depth==k)
{
cnt++;
return ;
}
rep(i,x,n+1) rep(j,1,n+1)
{
if(chk(i,j))
{
dfs(i,j,depth+1);
visr[i]=visc[j]=0;
}
}
}
int main(){
while(scanf("%d%d",&n,&k)&&(n!=-1||k!=-1))
{
rep(i,1,n+1) rep(j,1,n+1) scanf(" %c",&grid[i][j]);
cnt=0;
rep(i,1,n+1) rep(j,1,n+1)
{
if(grid[i][j]!='.')
{
memset(visr,0,sizeof(visr));
memset(visc,0,sizeof(visc));
dfs(i,j,1);
}
}
printf("%d\n",cnt);
}
return 0;
}