leetcode_733

Golang:

思路：这题使用BFS，每找到一个元素，就染色，染完色后，下次就不会找到这个染过色的元素了。所以要注意一下，如果新染的颜色和原来一样，就不染了，直接返回即可。

代码如下：

func floodFill(image [][]int, sr int, sc int, newColor int) [][]int {
    oldColor:=image[sr][sc]
    if newColor==oldColor{
        return image
    }
    var stack [][]int
    stack=append(stack, []int{sr,sc})
    for len(stack)!=0{
        tr:=stack[0][0]
        tc:=stack[0][1]
        image[tr][tc]=newColor
        stack=stack[1:]
        if tr>=1&&image[tr-1][tc]==oldColor {
            image[tr-1][tc]=newColor
            stack=append(stack, []int{tr-1,tc})
        }
        if tr<len(image)-1&&image[tr+1][tc]==oldColor {
            image[tr+1][tc]=newColor
            stack=append(stack, []int{tr+1,tc})
        }
        if tc>=1&&image[tr][tc-1]==oldColor {
            image[tr][tc-1]=newColor
            stack=append(stack, []int{tr,tc-1})
        }
        if tc<len(image[0])-1&&image[tr][tc+1]==oldColor {
            image[tr][tc+1]=newColor
            stack=append(stack, []int{tr,tc+1})
        }
    }
    return image
}

有疑问加站长微信联系（非本文作者）