剑指 offer 第 8 题:二叉树的下一个结点

2022-04-27  本文已影响0人  放开那个BUG

1、前言

题目描述

2、思路

示例图

节点(设为x)中序遍历的下一个节点有以下可能:

  • 若x有右子树。则x的下一个节点为x右子树最左侧节点。如,2的下一个节点为8。
  • 若x没有右子树,又分为2种情况。
    若x是父节点的左孩子。则x的父节点就是x的下一个节点。如,7的下一个节点是4。
    若x是父节点的右孩子。则沿着父节点向上,直到找到一个节点的父节点的左孩子是该节点,则该节点的父节点就是x的下一个节点。如,9的下一个节点是1。

3、代码

public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode) {
        // 如果有右子树,则找右子树的最左节点
        if(pNode.right != null){
            TreeLinkNode p = pNode.right;
            while(p.left != null){
                p = p.left;
            }
            return p;
        }
        
        TreeLinkNode p = pNode;
        // 如果没有右子树,则找到第一个当前节点是父节点左孩子的节点
        while(p.next != null){
            if(p.next.left == p){
                return p.next;
            }
            p = p.next;
        }
        return null;
    }
}
上一篇下一篇

猜你喜欢

热点阅读