常用算法代码模板与题单 | B - 搜索

这篇文档其实是我自己学习路上的一份“复习笔记”。之前看了不少教程，零零散散记了一堆代码片段和小技巧，时间久了就容易忘，翻起来也麻烦。所以干脆花时间整理了一下，把常用的模板、容易踩的坑、还有一些个人觉得挺实用的小技巧都归到了一起。

如果你也在学类似的东西，或许能帮你省下一些翻找的时间；如果有哪里写得不清楚或不对，也欢迎随时指正交流。

希望这份整理也能对你有一点点帮助～ ✨

【递归】

算法模板

void fun(x)
{
	if (..)
	{
		return;  // 递归退出条件
	}
	fun(y);  // 数据拆分
}

算法应用

【DFS - 一维】

算法模板

void dfs(int u)
{
    if (u==n)
	{  // 到达n层后结束搜索
        for (int i=0; i<n; i++)
		{
			// 按照题目要求执行，或者打印、计算等
        }
        return ;  // 一定要加这句返回
    }
	// 注意：DFS没有通用模板，每道题都有自己的顺序，这里需要根据每道题进行修改
    for (int i=1; i<=n; i++)
	{  // 每个位置可以遍历的内容，如这里是可以填入1~n的数字
        if (!st[i])
		{  // 判断状态，确定是否可以填入
            path[u] = i;
            st[i] = true;
            dfs(u+1);  // DFS搜索
            // path[u] = 0; // 可以不要这句，因为每次都被赋值
            st[i] = false;
        }
    }
}
// 从第0层开始遍历
dfs(0);

// 剪枝方法1
// 增加sum参数，直接传递当前已选数的总和，避免重复计算，从而提高效率
void dfs(int step, int sum)

// 剪枝方法2
// 只要满足一种方案就退出
void dfs(int step)
{
    if (flag) return;
}

算法应用

【DFS - 二维】

算法模板

算法应用

【BFS - 一维】

算法模板

// 该模板存在问题是队列长度只增不减，可使用vector方式保持队列长度始终在范围内
typedef pair<int, int> PII;  // 三个及以上的参数适合用结构体
PII q[N*N];
int d[N][N];  // 记录每个点到起点的距离
int dx[4]={-1,0,1,0};
int dy[4]={0,1,0,-1};
int bfs()
{
	int hh=0, tt=0;
	q[0] = {0, 0};
	memset(d, -1, sizeof(d));  // 初始化每个点到起点的距离
	d[0][0] = 0;  // 起点的距离为0
	while (hh<=tt)
	{
		auto t = q[hh++];
		for (int i=0; i<4; i++)
		{
			int x = t.first+dx[i], y = t.second+dy[i];
			if (x>=0 && x<n && y>=0 && y<m && g[x][y]==0 && d[x][y]==-1)
			{
				d[x][y] = d[t.first][t.second]+1;
				q[++tt] = {x,y};
			}
		}
	}
	return d[n-1][m-1];
}
bfs()  // 调用bfs()，返回最短路径

算法应用

【BFS - 二维】

算法模板

单源BFS

// 用vector改写
typedef pair<int, int> PII;  // 三个及以上的参数适合用结构体
queue<PII> q;
int d[N][N];  // 记录每个点到起点的距离，这里无需vis数组（dist[][]==-1表示没走过）
int dx[4]={-1, 1, 0, 0};
int dy[4]={0, 0, -1, 1};
void bfs()
{
	memset(d, -1, sizeof(d));  // 初始化每个点到起点的距离
	q.push({0,0});
	d[0][0] = 0;  // 起点的距离为0
	while (!q.empty())
	{
		int x = q.front().first, y = q.front().second;
		q.pop();
		for (int i=0; i<4; i++)
		{
			int nx = x+dx[i], ny = y+dy[i];
			if (nx<1 || nx>n || ny<1 || ny>n || g[nx][ny]==1 || d[nx][ny]!=-1)
				continue;
			d[nx][ny] = d[x][y]+1;
			q.push({nx, ny});
		}
	}
}
bfs()  // 调用bfs()，dist[i][j]中保存着最短路

多源BFS

// 与单源BFS不一样的地方在与需要将多个值压入队列
void bfs()
{
	// ...
	for (int i=1; i<=cur; i++)
		q.push(a[i]);
	// ...
}

算法应用

【IDA*】

算法模板

算法应用

【A*】

算法模板

算法应用

【折半搜索】

算法模板

DFS版折半搜索

void dfs(..., int type)  // type用于标记属于前半部分还是后半部分
{
	// 标准dfs过程
	if (...)
	{
		// 达到条件后的处理

		if (type==1) l[sum]++;  // 保存在前半部分
		else r[sum]++;  // 保存在后半部分
		return;
	}
	dfs(..., type);
	dfs(..., type);
}
unordered_map<int, int> l, r;  // 用map记录前半部分和后半部分
dfs(1, mid, 0, 1);  // 搜索前半部分
dfs(mid+1, n, 0, 2);  // 搜索后半部分

枚举版折半搜索

// 枚举前半部分
for (i : 前半部分)
{
	// 将结果存到map中
}
// 枚举后半部分
for (i : 后半部分)
{
	// 得到后半部分结果
	// 计算需要的前半部分的结果
	// 累加组合数（求和的过程）
}

算法应用

【迭代加深】

算法模板

算法应用

【舞蹈链】

算法模板

算法应用