二叉树中的一个节点的后继节点

二叉树中的一个节点的后继节点

这是第一次接触二叉树中后继节点这个概念，觉得很好玩就来看一看。所谓的后继节点就是中序遍历二叉树后这个节点后面的那个节点。

题目

二叉树的定义多了一个指向父节点的指针，根节点的父节点指针为空。

struct node {
    int Value
    *node Left
    *node Right
    *node Parent
}

随机给出一个节点，找出这个节点的后继节点。

方法

• 普通方法
• 进阶方法

普通方法

思路很简单，不断的找父节点，直到找到根节点，开始对当前二叉树进行中序遍历，在中序遍历的过程中比对是否为需要判断的节点。
这是很容易理解的一种思路，时间复杂度是O(N),N是二叉树节点个数。

进阶方法

既然后继节点是根据中序遍历判断的，那么直接找中序遍历的特点来解决问题。思路如下：

1. 最简单情况是node有右子树，那么后继节点就是右子树的最左节点。
2. node没有右子树：
   • 是父节点的左孩子，那么后继节点是父节点
   • 是父节点的右孩子，不断找父节点，直到找到s节点，s节点是父节点p的左孩子。那么p就是node节点的后继节点。（node是以p为根节点的二叉树的左子树中的最右节点）
3. 第2步没有满足的节点，证明当前节点是二叉树右子树的最右节点，即二叉树中序遍历的最后一个节点，无后继节点。

func getNextNode(*node n) {
    if n == nil {
        return nil
    }
    
    //情况1
    if n.Right != nil {
        return getMostLeft(n.Right)
    }
    
    //情况2和3，这个边界的处理做的比较好，处理了3中场景
    p := n.Parent
    for p != nil && p.Left != n {
        n = p
        p = n.Parent
    }
    
    return p
}

func getMostLeft(*node n) {
    if n == nil {
        return nil
    }
    
    for n.Left != nil {
        n = n.Left
    }
    
    return n
}

关于作者

大四学生一枚，分析数据结构，面试题，golang，C语言等知识。QQ交流群：521625004。微信公众号：后台技术栈。