算法提高课笔记
目录
定义
什么是 Flood Fill 算法？
字面意思理解：洪水覆盖
也就是说，下图的格子分为两大类，一类比较高一类比较低，现在从任意一处较低的格子开始灌水，下一次水将会覆盖它相邻的、较低的格子，依此类推
这就相当于是BFS的思想（也可以用DFS实现，但BFS不会出现爆栈的问题）
Flood Fill 算法可以在线性时间复杂度内，找到某个点所在的连通块
例题
池塘计数
农夫约翰有一片 N∗M 的矩形土地。
最近，由于降雨的原因，部分土地被水淹没了。
现在用一个字符矩阵来表示他的土地。
每个单元格内，如果包含雨水，则用"W"表示，如果不含雨水，则用"."表示。
现在，约翰想知道他的土地中形成了多少片池塘。
每组相连的积水单元格集合可以看作是一片池塘。
每个单元格视为与其上、下、左、右、左上、右上、左下、右下八个邻近单元格相连。
请你输出共有多少片池塘，即矩阵中共有多少片相连的"W"块。
输入格式
第一行包含两个整数 N 和 M。
接下来 N 行，每行包含 M 个字符，字符为"W"或"."，用以表示矩形土地的积水状况，字符之间没有空格。
输出格式
输出一个整数，表示池塘数目。
数据范围
1 ≤ N, M ≤ 1000
输入样例
cpp
10 12
W........WW.
.WWW.....WWW
....WW...WW.
.........WW.
.........W..
..W......W..
.W.W.....WW.
W.W.W.....W.
.W.W......W.
..W.......W.输出样例
cpp
3题意
模板题，问有多少连通块
思路
按顺序遍历图上每一个点，如果这个点是洼地且未被遍历过，就在答案的基础上加1，并且标记这个连通块的所有点，直到遍历完整个图
代码
cpp
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
#define ft first
#define sd second
const int N = 1010, M = N * N;
int n, m;
char g[N][N]; // 存图
queue<PII> q;
bool st[N][N]; // 判重
void bfs(int x, int y)
{
q.push({x, y});
st[x][y] = true;
while (q.size())
{
PII t = q.front();
q.pop();
for (int i = t.ft - 1; i <= t.ft + 1; i ++ )
for (int j = t.sd - 1; j <= t.sd + 1; j ++ )
{
if (i == t.ft && j == t.sd) continue; // 正好遍历到当前点
if (i < 0 || i >= n || j < 0 || j >= m) continue; // 位置不合法
if (g[i][j] == '.' || st[i][j]) continue; // 不是洼地或已遍历过
q.push({i, j});
st[i][j] = true;
}
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n >> m;
for (int i = 0; i < n; i ++ ) cin >> g[i];
int cnt = 0;
for (int i = 0; i < n; i ++ )
for (int j = 0; j < m; j ++ )
if (g[i][j] == 'W' && !st[i][j])
{
bfs(i, j);
cnt ++ ;
}
cout << cnt << '\n';
}城堡问题
1 2 3 4 5 6 7
#############################
1 # | # | # | | #
#####---#####---#---#####---#
2 # # | # # # # #
#---#####---#####---#####---#
3 # | | # # # # #
#---#########---#####---#---#
4 # # | | | | # #
#############################
(图 1)
# = Wall
| = No wall
- = No wall
方向：上北下南左西右东。图1是一个城堡的地形图。
请你编写一个程序，计算城堡一共有多少房间，最大的房间有多大。
城堡被分割成 m∗n 个方格区域，每个方格区域可以有0~4面墙。
注意：墙体厚度忽略不计。
输入格式
第一行包含两个整数 m 和 n，分别表示城堡南北方向的长度和东西方向的长度。
接下来 m 行，每行包含 n 个整数，每个整数都表示平面图对应位置的方块的墙的特征。
每个方块中墙的特征由数字 P 来描述，我们用1表示西墙，2表示北墙，4表示东墙，8表示南墙，P 为该方块包含墙的数字之和。
例如，如果一个方块的 P 为3，则 3 = 1 + 2，该方块包含西墙和北墙。
城堡的内墙被计算两次，方块(1,1)的南墙同时也是方块(2,1)的北墙。
输入的数据保证城堡至少有两个房间。
输出格式
共两行，第一行输出房间总数，第二行输出最大房间的面积（方块数）。
数据范围
1 ≤ m, n ≤ 50,
0 ≤ P ≤ 15
输入样例
cpp
4 7
11 6 11 6 3 10 6
7 9 6 13 5 15 5
1 10 12 7 13 7 5
13 11 10 8 10 12 13 输出样例
cpp
5
9题意
给出一个城堡中每个房间四面是否有墙壁，输出该城堡有多少个连通的房间，最大的连通块面积是多少
思路
和上一题基本一样，判断每个房间的四面墙壁情况，只是要额外加一个面积的计算
代码
cpp
#include <bits/stdc++.h>
using namespace std;
const int N = 55;
typedef pair<int, int> PII;
#define ft first
#define sd second
int n, m;
int g[N][N]; // 存图
queue<PII> q;
bool st[N][N]; // 判重
int bfs(int x, int y)
{
int area = 0;
int dx[4] = {0, -1, 0, 1}, dy[4] = {-1, 0, 1, 0};
q.push({x, y});
st[x][y] = true;
while (q.size())
{
PII t = q.front();
q.pop();
area ++ ;
for (int i = 0; i < 4; i ++ )
{
int a = t.ft + dx[i], b = t.sd + dy[i];
if (a < 0 || a >= n || b < 0 || b >= m) continue; // 位置不合法
if (st[a][b]) continue; // 已被遍历
if (g[t.ft][t.sd] >> i & 1) continue; // 这个方向是墙
q.push({a, b});
st[a][b] = true;
}
}
return area;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n >> m;
for (int i = 0; i < n; i ++ )
for (int j = 0; j < m; j ++ )
cin >> g[i][j];
int cnt = 0, area = 0;
for (int i = 0; i < n; i ++ )
for (int j = 0; j < m; j ++ )
if (!st[i][j])
{
area = max(area, bfs(i, j));
cnt ++ ;
}
cout << cnt << '\n' << area;
}山峰和山谷
FGD小朋友特别喜欢爬山，在爬山的时候他就在研究山峰和山谷。
为了能够对旅程有一个安排，他想知道山峰和山谷的数量。
给定一个地图，为FGD想要旅行的区域，地图被分为 n×n 的网格，每个格子 (i,j) 的高度 w(i,j) 是给定的。
若两个格子有公共顶点，那么它们就是相邻的格子，如与 (i,j) 相邻的格子有(i−1,j−1),(i−1,j),(i−1,j+1),(i,j−1),(i,j+1),(i+1,j−1),(i+1,j),(i+1,j+1)。
我们定义一个格子的集合 S 为山峰（山谷）当且仅当：
- S 的所有格子都有相同的高度。
- S 的所有格子都连通。
- 对于 s 属于 S，与 s 相邻的 s′ 不属于 S，都有 ws>ws′（山峰），或者 ws<ws′（山谷）。
- 如果周围不存在相邻区域，则同时将其视为山峰和山谷。
你的任务是，对于给定的地图，求出山峰和山谷的数量，如果所有格子都有相同的高度，那么整个地图即是山峰，又是山谷。
输入格式
第一行包含一个正整数 n，表示地图的大小。
接下来一个 n×n 的矩阵，表示地图上每个格子的高度 w。
输出格式
共一行，包含两个整数，表示山峰和山谷的数量。
数据范围
1 ≤ n ≤ 1000,
0 ≤ w ≤ 109
输入样例1
cpp
5
8 8 8 7 7
7 7 8 8 7
7 7 7 7 7
7 8 8 7 8
7 8 8 8 8输出样例1
cpp
2 1输入样例2
cpp
5
5 7 8 3 1
5 5 7 6 6
6 6 6 2 8
5 7 2 5 8
7 1 0 1 7输出样例2
cpp
3 3题意
给出不同地块的高度，高度相同的连通块周边高度都小于等于该连通块高度，那么该连通块为山峰，反之为山谷，问山峰山谷数量
思路
依旧是遍历每一个点，如果没有遍历过该点，就遍历该点所在连通块，直到找到连通块边界，再判断这个连通块是否为山峰或山谷
代码
cpp
#include <bits/stdc++.h>
using namespace std;
const int N = 1010, M = N * N;
typedef pair<int, int> PII;
#define ft first
#define sd second
int n;
int h[N][N];
queue<PII> q;
bool st[N][N];
void bfs(int x, int y, bool& has_higher, bool& has_lower)
{
q.push({x, y});
st[x][y] = true;
while (q.size())
{
PII t = q.front();
q.pop();
for (int i = t.ft - 1; i <= t.ft + 1; i ++ )
for (int j = t.sd - 1; j <= t.sd + 1; j ++ )
{
if (i == t.ft && j == t.sd) continue; // 遍历到自身
if (i < 0 || i >= n || j < 0 || j >= n) continue; // 位置不合法
if (h[i][j] != h[t.ft][t.sd]) // 边界
{
if (h[i][j] > h[t.ft][t.sd]) has_higher = true;
else has_lower = true;
}
else if (!st[i][j]) // 非边界且未遍历
{
q.push({i, j});
st[i][j] = true;
}
}
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n;
for (int i = 0; i < n; i ++ )
for (int j = 0; j < n; j ++ )
cin >> h[i][j];
int peak = 0, valley = 0;
for (int i = 0; i< n; i ++ )
for (int j = 0; j < n; j ++ )
if (!st[i][j])
{
bool has_higher = false, has_lower = false;
bfs(i, j, has_higher, has_lower);
if (!has_higher) peak ++ ;
if (!has_lower) valley ++ ;
}
cout << peak << ' ' << valley << '\n';
}