文章目录
- 前言
- FloodFill算法简介
- 题目描述
- 算法原理
- 代码实现——BFS
- C++
- Java
前言
大家好啊,今天就正式开始我们的BFS专题了,觉得有用的朋友给个三连呗。
FloodFill算法简介
中文:洪水灌溉
举个例子,正数为凸起的山峰,负数为盆地,洪水冲过这片土地就会将这些具有相同性质的联通块(在本例中为盆地)灌溉。
题目描述
题目链接:733.图像渲染
简单来说就是给我们一个起点坐标,让我们从上下左右四个方向去找相同像素点的坐标。
算法原理
可以利用深搜或者宽搜,遍历到与该点相连的所有像素相同的点,然后将其修改成指定的像素值即可。
本篇博客使用的是BFS,每条斜线代表每一层遍历。
代码实现——BFS
C++
class Solution {typedef pair<int, int> PII;//技巧一int dx[4] = {0, 0, 1, -1};//技巧二:对应上下左右四个方向的坐标int dy[4] = {1, -1, 0, 0};public:vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc,int color) {int prev = image[sr][sc];if (prev == color)return image; // 处理边界情况int m = image.size(), n = image[0].size();queue<PII> q;q.push({sr, sc});while (!q.empty()) {auto [a, b] = q.front();image[a][b] = color;q.pop();for (int i = 0; i < 4; ++i) {int x = a + dx[i], y = b + dy[i];if (x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev) {q.push({x, y});}}}return image;}
};
Java
class Solution {int[] dx = { 0, 0, 1, -1 };int[] dy = { 1, -1, 0, 0 };public int[][] floodFill(int[][] image, int sr, int sc, int color) {int prev = image[sr][sc]; // 统计刚开始的颜⾊if (prev == color)return image; // 处理边界情况int m = image.length, n = image[0].length;Queue<int[]> q = new LinkedList<>();q.add(new int[] { sr, sc });while (!q.isEmpty()) {int[] t = q.poll();int a = t[0], b = t[1];image[a][b] = color;// 上下左右四个⽅向for (int i = 0; i < 4; i++) {int x = a + dx[i], y = b + dy[i];if (x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev) {q.add(new int[] { x, y });}}}return image;}
}