分享
leetcode_542
淳属虚构 · · 1005 次点击 · · 开始浏览这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。
Golang:
思路:多源BFS
代码如下:
func updateMatrix(matrix [][]int) [][]int {
res:=make([][]int,len(matrix))
for k,_:=range res{
res[k]=make([]int,len(matrix[0]))
copy(res[k],matrix[k])
}
var stack [][]int
for i:=0;i<len(matrix);i++{
for j:=0;j<len(matrix[0]);j++{
if matrix[i][j]!=0{
stack=append(stack,[]int{i,j})
}
}
}
nums:=1
for len(stack)!=0{
temp:=stack
stack=nil
var b0 [][]int
for _,v:=range temp{
if checkBorder(matrix,v[0],v[1]){
b0=append(b0,[]int{v[0],v[1]})
}else{
stack=append(stack,[]int{v[0],v[1]})
}
}
for _,v:=range b0{
matrix[v[0]][v[1]]=0
res[v[0]][v[1]]=nums
}
nums++
}
return res
}
func checkBorder(mat [][]int,i,j int) bool{
if (i+1<len(mat)&&mat[i+1][j]==0)||(i-1>=0&&mat[i-1][j]==0)||(j+1<len(mat[0])&&mat[i][j+1]==0)||(j-1>=0&&mat[i][j-1]==0){
return true
}
return false
}
有疑问加站长微信联系(非本文作者)
入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889
关注微信1005 次点击
添加一条新回复
(您需要 后才能回复 没有账号 ?)
- 请尽量让自己的回复能够对别人有帮助
- 支持 Markdown 格式, **粗体**、~~删除线~~、
`单行代码` - 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
- 图片支持拖拽、截图粘贴等方式上传
收入到我管理的专栏 新建专栏
Golang:
思路:多源BFS
代码如下:
func updateMatrix(matrix [][]int) [][]int {
res:=make([][]int,len(matrix))
for k,_:=range res{
res[k]=make([]int,len(matrix[0]))
copy(res[k],matrix[k])
}
var stack [][]int
for i:=0;i<len(matrix);i++{
for j:=0;j<len(matrix[0]);j++{
if matrix[i][j]!=0{
stack=append(stack,[]int{i,j})
}
}
}
nums:=1
for len(stack)!=0{
temp:=stack
stack=nil
var b0 [][]int
for _,v:=range temp{
if checkBorder(matrix,v[0],v[1]){
b0=append(b0,[]int{v[0],v[1]})
}else{
stack=append(stack,[]int{v[0],v[1]})
}
}
for _,v:=range b0{
matrix[v[0]][v[1]]=0
res[v[0]][v[1]]=nums
}
nums++
}
return res
}
func checkBorder(mat [][]int,i,j int) bool{
if (i+1<len(mat)&&mat[i+1][j]==0)||(i-1>=0&&mat[i-1][j]==0)||(j+1<len(mat[0])&&mat[i][j+1]==0)||(j-1>=0&&mat[i][j-1]==0){
return true
}
return false
}