中序遍历

gdis5251 · Jul 23, 2021 · 1f8417b · 1f8417b
1 parent ba90c1c
commit 1f8417b
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/精选/04.06.后继者/main.go b/精选/04.06.后继者/main.go
@@ -0,0 +1,34 @@
+package main
+
+type TreeNode struct {
+	Val   int
+	Left  *TreeNode
+	Right *TreeNode
+}
+
+var resArr = make([]*TreeNode, 0)
+
+func inorderSuccessor(root *TreeNode, p *TreeNode) *TreeNode {
+	if root == nil {
+		return nil
+	}
+
+	midOrder(root)
+
+	for index := range resArr {
+		if resArr[index] == p && index < len(resArr) - 1 {
+			return resArr[index + 1]
+		}
+	}
+	return nil
+}
+
+func midOrder(root *TreeNode) {
+	if root == nil {
+		return
+	}
+
+	midOrder(root.Left)
+	resArr = append(resArr, root)
+	midOrder(root.Right)
+}