POJ 1321 棋盘问题(八皇后升级版)
思路
跟八皇后一样的做法,八皇后问题解法:
POJ 2754八皇后问题 DFS
DFS函数有一些区别:
八皇后:棋子数=棋盘行数
棋盘问题:棋子数<=棋盘行数
所以棋盘问题在以第a行为起点搜索完以后,需要再以第a+1行为起点搜索
另外要注意:两种边界条件的先后顺序
(1)得到解 ;要先于(2)越界;
否则会造成最后一行搜到的解没有被计数
DFS函数代码:
void dfs(int i)
{
int j;
//1.边界
//判断解增加要在边界返回前面
if(put==k)
{
sum++;
return;
}
if(i>=n)
return;
//2.遍历和递归
for(j=0;j<n;j++)
{
if(block[i][j]==0)
{
if(!used[j])
{
used[j]=1;
put++;
dfs(i+1);
used[j]=0;
put--;
}
}
}
//3.递归下一行
dfs(i+1);//从这一行开始找完还要从下一行开始找!与八皇后最重要的区别
}
AC代码
#include <bits/stdc++.h>
using namespace std;
int block[8][8];
int sum,put;
int n,k;
bool used[8];
void dfs(int i)
{
int j;
//判断解增加要在边界返回前面
if(put==k)
{
sum++;
return;
}
if(i>=n)
return;
for(j=0;j<n;j++)
{
if(block[i][j]==0)
{
if(!used[j])
{
used[j]=1;
put++;
dfs(i+1);
used[j]=0;
put--;
}
}
}
dfs(i+1);//从这一行开始找完还要从下一行开始找!
}
int main()
{
while((scanf("%d%d",&n,&k))&&(n+k)!=-2)
{
getchar();
memset(block,-1,sizeof(block));
char in;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%c",&in);
if(in=='#')
{
block[i][j]=0;
}
}
getchar();
}
sum=0;
put=0;
memset(used,0,sizeof(used));
dfs(0);
printf("%d\n",sum);
}
return 0;
}