分享
leetcode_733
淳属虚构 · · 1004 次点击 · · 开始浏览这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。
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
}
有疑问加站长微信联系(非本文作者)
入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889
关注微信1004 次点击
添加一条新回复
(您需要 后才能回复 没有账号 ?)
- 请尽量让自己的回复能够对别人有帮助
- 支持 Markdown 格式, **粗体**、~~删除线~~、
`单行代码` - 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
- 图片支持拖拽、截图粘贴等方式上传
收入到我管理的专栏 新建专栏
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
}