问题描述:
解题思路:
典型的dfs问题。按dfs模板写,注意出口,改变状态,递归,恢复状态。
注意点:
1.全局变量定义了就不要再定义局部变量,否者会导致自定义函数传值错误,例:全局变量n,如果定义局部变量再main中,则main里的n不会在dfs()里被使用。
2.循环条件如果有两个条件,则需要用&&或||连接。
题解:
#include<bits/stdc++.h>
using namespace std;
const int N = 15;
int vis[N][N], ans = 0, n;
void dfs(int dep){ // 找到dep行空位置的重复操作
// 出口,如果到不了最终的话(即在某一行早不到空的位置放皇后),当遍历完一行的全部元素也会退出
if(dep == n + 1)
{
ans++;
return;
}
for(int i = 1; i <= n; i++) // 横向遍历某一行全部元素
{
if(vis[dep][i])continue; // 如果被标记则跳过这个位置
// 改变状态:在该行的没被占用的位置按规则改变状态(_j初始为i)
for(int _i = 1; _i <= n; _i++)vis[_i][i]++;
for(int _i = dep, _j = i; _i >= 1&&_j >= 1; _i--, _j--)vis[_i][_j]++;
for(int _i = dep, _j = i; _i >= 1&& _j <= n; _i--, _j++)vis[_i][_j]++;
for(int _i = dep, _j = i; _i <= n&&_j >= 1; _i++, _j--)vis[_i][_j]++;
for(int _i = dep, _j = i; _i <= n&& _j <= n; _i++, _j++)vis[_i][_j]++;
dfs(dep + 1); // 往深度递归
// 恢复状态:执行完一个放置方法后恢复
for(int _i = 1; _i <= n; _i++)vis[_i][i]--;
for(int _i = dep, _j = i; _i >= 1&& _j >= 1; _i--, _j--)vis[_i][_j]--;
for(int _i = dep, _j = i; _i >= 1&& _j <= n; _i--, _j++)vis[_i][_j]--;
for(int _i = dep, _j = i; _i <= n&&_j >= 1; _i++, _j--)vis[_i][_j]--;
for(int _i = dep, _j = i; _i <= n&& _j <= n; _i++, _j++)vis[_i][_j]--;
}
}
int main(){
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
cin >> n;
dfs(1); // 1开始
cout << ans << '\n';
return 0;
}
知识点:dfs(深度优先搜索)