题解 | #牛群的树形结构重建#
牛群的树形结构重建
https://www.nowcoder.com/practice/bcabc826e1664316b42797aff48e5153?tpId=354&tqId=10591490&ru=/exam/oj/ta&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj%2Fta%3FtpId%3D354
知识点:
树的中序遍历和树的后序遍历
解题思路:
通过后序遍历来确定当前节点,在用后序遍历的val在中序遍历的位子来分割左右子树
语言:
Golang
package main
import . "nc_tools"
/*
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param inOrder int整型一维数组
* @param postOrder int整型一维数组
* @return TreeNode类
*/
func buildTree( inOrder []int , postOrder []int ) *TreeNode {
// write code here
index:=map[int]int{}
for i,v:=range inOrder{
index[v]=i
}
return build(inOrder,0,len(inOrder)-1,postOrder,0,len(postOrder)-1,index)
}
func build(inOrder []int,inStart,inEnd int,postOrder []int,poStart,poEnd int,index map[int]int) *TreeNode{
if inStart>inEnd{
return nil
}
rootVal:= postOrder[poEnd]
cur:=index[rootVal]
size:=cur-inStart
root:=&TreeNode{Val: rootVal}
root.Left= build(inOrder,inStart,cur-1,postOrder,poStart,poStart+size-1,index)
root.Right= build(inOrder,cur+1,inEnd,postOrder,poStart+size,poEnd-1,index)
return root
}
海康威视公司氛围 1020人发布