题目概述
- 算法说明
给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。
- 测试用例
案例不太容易模拟，建议直接在牛客或者类似 oj 系统上测试。
解析&参考答案
- 解析
分两种情况：
1）节点有右子树，那么其下一个节点就是右子树的最左边的节点；
2）没有右子节点，需要判断其有没有父节点，且是不是父节点的左子节点，若是则找到目标节点；否则不断地向上找。
- 参考答案
vim jz57.go
package main
type TreeLinkNode struct {
Val int
Left *TreeLinkNode
Right *TreeLinkNode
Next *TreeLinkNode
}
func GetNext(pNode *TreeLinkNode) *TreeLinkNode {
if pNode == nil {
return pNode
}
if pNode.Right != nil {
root := pNode.Right
for root.Left != nil {
root = root.Left
}
return root
}
for pNode.Next != nil {
root := pNode.Next
if root.Left == pNode {
return root
} else {
pNode = pNode.Next // 向上找其根节点
}
}
return nil
}
func main() {
}
注意事项
- 当节点没有右子节点的时候，需要依次向上寻找目标节点。
说明
- 当前使用 go1.15.8
- 参考 牛客网--剑指offer
标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。
注意!!!
- 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
- 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
- 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解
