树09--二叉树的下一个结点

题目概述

• 算法说明
  给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的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() {

}

注意事项

1. 当节点没有右子节点的时候，需要依次向上寻找目标节点。

说明

1. 当前使用 go1.15.8
2. 参考 牛客网--剑指offer
   标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

• 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
• 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解