题目:
有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i][j] 表示该图画的像素值大小。
你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。
为了完成 上色工作 ,从初始像素开始,记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为 newColor 。
最后返回 经过上色渲染后的图像 。
算法:
方法一:dfs
import "fmt"
func floodFill(image [][]int, sr int, sc int, color int) [][]int {
m := len(image)
n := len(image[0])
var dfs func(i, j int, targetColor int)
dfs = func(i, j, targetColor int) {
if 0 <= i && i < m && 0 <= j && j < n && image[i][j] == targetColor {
image[i][j] = -1
dfs(i-1, j, targetColor)
dfs(i+1, j, targetColor)
dfs(i, j-1, targetColor)
dfs(i, j+1, targetColor)
}
}
dfs(sr, sc, image[sr][sc])
for i := range image {
for j := range image[i] {
if image[i][j] == -1 {
image[i][j] = color
}
}
}
return image
}