DFS序列问题,枚举搜索
DFS序列问题,枚举搜索。
题目①:棋盘
题目②:8皇后
题目③:踩方格
深度优先搜索可以用来遍历所有选择,找到所有的终结状态,从而找
到所有的解。在此基础上,我们就能轻松回答两类问题:(1)一共有多少可行解。
(2)哪个是最优解。
POJ-1321棋盘问题
Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
每组数据的第一行是两个正整数,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
代码①:
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 10
/*思路
二维数组a 表示棋盘
一维数组b 表示当前列是否有棋子
dfs 搜索下一步走法
结束条件step==k
搜索中枚举下一行 判断当前列是否有棋子
*/
using namespace std;
int n,k,ans;bool a[maxn][maxn],b[maxn];
inline int read()
{
bool f=0;char ch=getchar();
while (ch<'0'||ch>'8') f^=!(ch^45),ch=getchar();
return f? (~(ch^48))+1:ch^48;
}
inline bool fchar()
{
char ch=getchar();
while (ch^'#'&&ch^'.') ch=getchar();
return ch^'.';
}
void DFS(int l,int stp)
{
if (stp==k){ans++;return;}
for (int i=l+1;i<=n;i++)
for (int j=1;j<=n;j++) if (a[i][j]&&!b[j])
b[j]=1,DFS(i,stp+1),b[j]=0;
}
int main()
{
while ((n=read())!=-1&&(k=read())!=-1)
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));ans=0;
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++) a[i][j]=fchar();
DFS(0,0);
printf("%d\n",ans);
}
return 0;
}
代码②:
1 #include<iostream>
2 using namespace std;
3
4 long long ans = 0;
5 int n,k;
6 char a[10][10];
7
8 bool canDo(int x,int y){
9 return a[x][y] == '.' && x>=1 && x<=n && y>=1 && y<=n;
10 }
11
12 bool attrack(int x,int y){
13 for(int i=1;i<=n;i++){
14 if(a[x][i]=='1') return true;
15 }
16 for(int j=1;j<=n;j++){
17 if(a[j][y]=='1') return true;
18 }
19 return false;
20 }
21
22 void dfs(int x,int y,int k){
23 if(k<=0){
24 ans++;
25 return;
26 }
27 for(int i=1;i<=n;i++){
28 for(int j=1;j<=n;j++){
29 if(canDo(i,j) && !attrack(i,j)){
30 a[i][j] = '1';
31 k--;
32 dfs(i,j,k);
33 k++;
34 a[i][j] = '.';
35 }
36 }
37 }
38
39 }
40
41 int main(){
42 while(cin>>n>>k && n!=-1 && k!=-1){
43 ans = 0;
44 for(int i=1;i<=n;i++){
45 cin>>a[i];
46 }
47 dfs(1,1,k);
48
49 cout<<ans<<endl;
50 }
51 return 0;
52 }
8皇后问题:
1 #include<iostream>
2 using namespace std;
3
4 int ans=0;
5 int a[8];
6
7 //attrack判断能否攻击到
8 bool attrack(int x0,int step0,int x1,int step1){
9 if(x0 == x1 || step0 == step1 || x0-step0 == x1-step1 || x0+step0 == x1+step1){
10 return true;
11 }
12 return false;
13 }
14
15 void dfs(int x){
16
17 //dfs终止条件 x==8时即成功一种方案
18 if(x==8){
19 ans++;
20 return;
21 }
22
23 //当前第x行,使用i遍历从位置1~8,找到合适位置
24 for(int i=0;i<8;i++){
25 bool ok = true;
26 //前x-1行各个元素放置位置与当前枚举第x行放在第i列攻击比较
27 for(int j=0;j<x;j++){
28 if(attrack(j,a[j],x,i)){
29 ok = false;
30 break;
31 }
32 }
33 //如果不被攻击 dfs暴力搜索枚举下一行的方案
34 if(ok){
35 a[x] = i;
36 dfs(x+1);
37 }
38 }
39 }
40
41 int main(){
42 dfs(0);//从0开始暴力搜索
43 cout<<ans<<endl;
44 return 0;
45 }
百练4982踩方格
代码:
1 #include<iostream>
2 using namespace std;
3
4 int n,ans=0;
5 int visited[10001][10001] = {0};
6 int _map[51][51];
7
8 void dfs(int x,int y,int sum){
9 if(sum == n){
10 ans++;
11 return;
12 }
13 if( x+1<=50 && !visited[x+1][y] ){
14 visited[x+1][y] = 1;
15 dfs(x+1,y,sum+1);
16 visited[x+1][y] = 0;
17 }
18 if( x-1>=1 && !visited[x-1][y] ){
19 visited[x-1][y] = 1;
20 dfs(x-1,y,sum+1);
21 visited[x-1][y] = 0;
22 }
23 if(y+1<=50 && !visited[x][y+1] ){
24 visited[x][y+1] = 1;
25 dfs(x,y+1,sum+1);
26 visited[x][y+1] = 0;
27 }
28 }
29
30 int main(){
31 cin>>n;
32 ans = 0;
33 dfs(30,1,0);
34 cout<<ans<<endl;
35 }

浙公网安备 33010602011771号