Go语言实现：【剑指offer】二叉树的下一个结点

少侠请留步。

于 2019-12-08 20:10:59 发布

阅读量258

点赞数

分类专栏：算法文章标签： Golang

本文链接：https://blog.youkuaiyun.com/dubinyang/article/details/103448423

版权

算法专栏收录该内容

69 篇文章

订阅专栏

本文介绍了一种算法，用于在给定二叉树及其一个节点的情况下，找到该节点在中序遍历顺序下的下一个节点。算法首先检查当前节点是否有右子树，如果有，则返回右子树中最左侧的节点；如果没有右子树，则向上遍历到当前节点是其父节点左子节点的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

该题目来源于牛客网《剑指offer》专题。

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。

注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

Go语言实现：

type TreeNode struct {
   Val    int
   Left   *TreeNode
   Right  *TreeNode
   Parent *TreeNode
}

func getNext(node *TreeNode) *TreeNode {
   if node == nil {
      return nil
   }

   //node有右子树，则取右子树的最左结点
   if node.Right != nil {
      next := node.Right
      for next.Left != nil {
         next = next.Left
      }
      return next
   } else {
      //node没有右子树且存在父节点
      for node.Parent != nil {
         //node为父结点的左孩子，则取node的父结点
         if node.Parent.Left == node {
            return node.Parent
         } else {
            //node为父结点的右孩子，则取node.P遍历，直到node.P是node.P.P的左孩子为止
            node = node.Parent
         }
      }
   }
   return nil
}