C++ | DFS搜索（回溯法）学习分享

DFS存储遍历的图或树，回溯法不存储，下面不严格区分

DFS实质就是递归，树或图找一个点满足条件然后接着递归找相邻到底点。DFS搜索就是在枚举这些变量的所有组合。

回溯法（多用于排列型和子集型）模版：

• 下面是一个排列型搜索树
• vis[i]表示数字i是否使用过，也常用来表示某个元素是否使用过
• 子集型搜索树模版类似，就死往下走的时候只有两条边，表示“选或不选当前这个元素”

求 1~n 的全排列

int a[N];
bool vis[N];

void dfs(int dep) {
	//能到n+1层说明前n层已经遍历完成了
	if (dep == n + 1)
	{
		for (int i = 1; i <= n; i++) cout << a[i] << ' ';
		cout << '\n';
		return;//不带参数就是结束程序
	}
	for (int i = 1; i <= n; i ++) 
	{
		//排除不合法的路径
		if (vis[i]) continue;
		//修改状态
		vis[i] = true;
		a[dep] = i;
		
		dfs(dep + 1);
		
		//恢复现场
		vis[i] = false;
		a[dep] = 0;//可省
		
	}
} 

 

题解：将皇后对应的米字型区域设置成“禁区”，后面的皇后就不能放在“禁区”内，回溯时将禁区取消掉。 需要用int数组来表示“一个位置被多少个皇后占了”，当占数为0的时候表示“禁区解除”。

#include<bits/stdc++.h>
using namespace std;
const int N = 15;
int a[N], vis[N][N];
int n, ans;

void dfs(int dep) 
{
  if (dep == n + 1) 
  {
    ans ++;
    return;
  }

  for (int i = 1; i <= n; i ++)
  {
    if (vis[dep][i]) continue;
    //修改状态
    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 <= n && _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 <= 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 <= n && _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 <= n; _i ++, _j ++) vis[_i][_j] --;//↘
  }
}

int main() {
  ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
  cin >> n;
  dfs(1);
  cout << ans << '\n';

  return 0;
}

 

题解：可以看成一个点，路径是有向或环

跑dfs求最大的环，用时间戳（dfn），将走过的地方标记一个时间戳（即第几步走到的）。

在后续搜索中，每次开始时更新最小时间戳（否则会认为7、4之间有环），如果走到已经走到过的点就必须停下。

***全局变量的初始值为0***

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 9;
int n, a[N], dfn[N], idx, mindfn;

int dfs(int x) 
{
    //更新时间戳，就是说给了一个新的时间戳
	dfn[x] = ++ idx;

	if (dfn[a[x]])//如果指向的下一个小朋友有时间戳说明结束，如果是一个大于等于最小时间戳的就返回一个有效值——用当前时间戳减去下一个小朋友的时间戳
    {
		if (dfn[a[x]] >= mindfn) return dfn[x] - dfn[a[x]] + 1;

		return 0;//递归出口
	}

	return dfs(a[x]);

}

int main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	cin >> n;
	for (int i = 1; i <= n; i ++) cin >> a[i];
	int ans = 0;
	for (int i = 1; i <= n; i ++) 
    {
		if (!dfn[i]) //如果没有时间戳，说明这个还没有遍历过
        {
			mindfn = idx + 1;//idx表示此时的时间
			ans = max(ans, dfs(i));
		}
	}

	cout << ans << '\n';

	return 0;

}

lanqiao OJ 178全球变暖题解：

给一块连接着的土地染成一种颜色（只要该点没有颜色且是陆地就进bfs找一下上下左右，同理没颜色且是陆地就染同一种颜色，递归结束的标志就是说他的相邻是海了，注意也要判断说他的周围不能有颜色）。在淹没后只用判断这点的土地颜色是否存在，不存在ans加一，要注意把这点的vis数组改成true就ok了。                                                                              

坐标拆成两个一维数组的思想，其实就是表示一个点的上下左右

dx[0, 0, 1, -1];

dy[1, -1, 0, 0];

 我在这道题中犯的一个错误是在判断是否是海洋的时候，将等于号打成了赋值号，写题时除了思维还要认真缜密。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e3 + 5;

char mp[N][N];
int n, scc, col[N][N];

bool vis[N * N];//表示一整块陆地的一种颜色

int dx[] = {0, 0, 1, -1};
int dy[] = {1, -1, 0, 0};

void dfs(int x, int y) 
{
	col[x][y] = scc;
	for (int i = 0; i < 4; i ++) 
    {
		//nx mains  next x;
		int nx = x + dx[i], ny = y + dy[i];
		//if have color didnot rerun again
		if (col[nx][ny] || mp[nx][ny] == '.') continue;

		dfs(nx, ny);
	}
}

int main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

	cin >> n;
	for (int i = 1; i <= n; i ++) cin >> mp[i] + 1;

	for (int i = 1; i <= n; i ++) 
    {
		for (int j = 1; j <= n; j ++) 
        {
			if (col[i][j] || mp[i][j] == '.') continue;

			scc ++;//scc表示当前的颜色编号

			dfs(i, j);
		}
	}
	int ans = 0;
	for (int i = 1; i <= n; i ++) 
    {
		for (int j = 1; j <= n; j ++) 
        {
			if (mp[i][j] == '.') continue;

			bool tag = true;
			for (int k = 0; k < 4; k ++) 
            {
				int x = i + dx[k], y = j + dy[k];
				if (mp[x][y] == '.') tag = false;
			}

			if (tag) 
            {
				if (!vis[col[i][j]]) ans ++;
				vis[col[i][j]] = true;
			}
		}
	}

	cout << scc - ans << '\n';

	return 0;
}