【golang必备算法】二叉树构造篇

二叉树构造篇

889. 根据前序和后序遍历构造二叉树

给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

func buildTree(preorder []int, inorder []int) *TreeNode {if len(preorder)==0 || len(inorder)==0{return nil}index := find(preorder,inorder) //找到分界点return &TreeNode{Val : preorder[0],Left : buildTree(preorder[1:len(inorder[:index])+1],inorder[:index]),Right : buildTree(preorder[len(inorder[:index])+1:],inorder[index+1:]),}
}func find(preorder,inorder []int)int{for i:=0;i<len(inorder);i++{if preorder[0]==inorder[i]{return i}}return -1
}

关键代码解释:
第8行:

 Left : buildTree(preorder[1:len(inorder[:index])+1],inorder[:index]),
  • 中序遍历直接从index分,而前序遍历可以根据中序遍历长度推断出来,因为同一个的树前序、中序数量是一样的。
  • 这里加1是因为第一个是根节点,左子树的前序遍历不包括这个所以加1

第9行:

 Right : buildTree(preorder[len(inorder[:index])+1:],inorder[index+1:])
  • 中序遍历从index分,前序遍历结果是根节点、左子树递归遍历、右子树递归遍历
  • preorder[:len(inorder[:index])+1]是根节点加左子树递归遍历结果
  • 所以右子树前序遍历就是preorder[len(inorder[:index])+1:]

106. 从中序与后序遍历序列构造二叉树

与前序和后序遍历构造二叉树类似,都会用到一个很重的点,就是中序数组大小一定是和后序数组的大小相同的

func buildTree(inorder []int, postorder []int) *TreeNode {if len(inorder)==0 || len(postorder)==0{return nil}index := find(inorder,postorder)return &TreeNode{Val : postorder[len(postorder)-1],Left :buildTree(inorder[:index],postorder[:index]),// 同一个的树后序、中序数量是一样的Right :buildTree(inorder[index+1:],postorder[index:len(postorder)-1]),}
}func find(inorder []int,postorder []int)int{for i:=0;i<len(inorder);i++{if inorder[i] == postorder[len(postorder)-1]{return i}}return -1
}

第9行 Right 这里:

 Right :buildTree(inorder[index+1:],postorder[index:len(postorder)-1]),
  • 首先后序遍历数组postorder最后一个元素不要,它是根节点,还剩下左子树后序遍历数组+右子树后序遍历数组
  • 其次因为中序数组大小一定是和后序数组的大小相同,左子树中序遍历数组为inorder[:index]
  • 所以右子树后序遍历数组为postorder[index:len(postorder)-1])

参考:

代码随想录

labuladong


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部