bfs 733

Author: yiGmMk

go/bfs-dfs/733/733.图像渲染.go

@@ -0,0 +1,99 @@
+/*
+ * @lc app=leetcode.cn id=733 lang=golang
+ *
+ * [733] 图像渲染
+ *
+ * https://leetcode.cn/problems/flood-fill/description/
+ *
+ * algorithms
+ * Easy (58.56%)
+ * Likes:    415
+ * Dislikes: 0
+ * Total Accepted:    147.9K
+ * Total Submissions: 252.2K
+ * Testcase Example:  '[[1,1,1],[1,1,0],[1,0,1]]\n1\n1\n2'
+ *
+ * 有一幅以 m x n 的二维整数数组表示的图画 image ，其中 image[i][j] 表示该图画的像素值大小。
+ *
+ * 你也被给予三个整数 sr ,  sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。
+ *
+ * 为了完成 上色工作 ，从初始像素开始，记录初始坐标的 上下左右四个方向上
+ * 像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上
+ * 像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为 newColor 。
+ *
+ * 最后返回 经过上色渲染后的图像 。
+ *
+ *
+ *
+ * 示例 1:
+ *
+ *
+ *
+ *
+ * 输入: image = [[1,1,1],[1,1,0],[1,0,1]]，sr = 1, sc = 1, newColor = 2
+ * 输出: [[2,2,2],[2,2,0],[2,0,1]]
+ * 解析: 在图像的正中间，(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。
+ * 注意，右下角的像素没有更改为2，因为它不是在上下左右四个方向上与初始点相连的像素点。
+ *
+ *
+ * 示例 2:
+ *
+ *
+ * 输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
+ * 输出: [[2,2,2],[2,2,2]]
+ *
+ *
+ *
+ *
+ * 提示:
+ *
+ *
+ * m == image.length
+ * n == image[i].length
+ * 1 <= m, n <= 50
+ * 0 <= image[i][j], newColor < 2^16
+ * 0 <= sr < m
+ * 0 <= sc < n
+ *
+ *
+ */
+package jzoffer
+
+// @lc code=start
+func floodFill(image [][]int, sr int, sc int, color int) [][]int {
+	type point struct {
+		x, y int
+	}
+	preColor := image[sr][sc] // 记录初始颜色
+
+	queue := []point{{sr, sc}}
+	visited := make(map[point]bool)
+
+	// 查找邻居
+	neighbors := func(p point) (points []point) {
+		ps := []point{{p.x, p.y - 1}, {p.x, p.y + 1}, {p.x - 1, p.y}, {p.x + 1, p.y}}
+		for _, v := range ps {
+			if v.x >= 0 && v.x < len(image) &&
+				v.y >= 0 && v.y < len(image[0]) {
+				if !visited[v] && image[v.x][v.y] == preColor {
+					points = append(points, v)
+				}
+			}
+		}
+		return
+	}
+	for len(queue) > 0 {
+		top := queue[len(queue)-1]
+		queue = queue[:len(queue)-1]
+		if visited[top] {
+			continue
+		}
+		image[top.x][top.y] = color
+		visited[top] = true
+		queue = append(queue, neighbors(top)...)
+
+	}
+	return image
+}
+
+// @lc code=end

go/bfs-dfs/733/733_test.go

@@ -0,0 +1,14 @@
+package jzoffer
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test733(t *testing.T) {
+	// [[1,1,1],[1,1,0],[1,0,1]]
+	// 1
+	// 1
+	// 2
+	fmt.Println(floodFill([][]int{{1, 1, 1}, {1, 1, 0}, {1, 0, 1}}, 1, 1, 2))
+}