五分钟掌握:二叉树的下一个节点

Posted 一个搬砖的胖子

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了五分钟掌握:二叉树的下一个节点相关的知识,希望对你有一定的参考价值。

前言

今天补充的题目虽是剑指offer的题,但Leetcode上没有。

在各大公司面试都容易被问,加在补充题也是希望能引起大家的注意。

一起花5min学习一下这道题吧。

题目描述

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

注意,树中的结点不仅包含左右子结点,而且包含指向父结点的指针。

题目分析

以下解析参考牛客@刘豪杰思路非常清晰。

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

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

我个人觉得这题并不简单。之前我也做过这道题,但是这两天去做,又遗漏了一种情况。看来重复刷题真的很重要!

掌握了的话,提交代码试试~

参考代码

//参考牛客@weizier的JAVA代码
/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next; //父节点
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(pNode->right) { //如果有右子树,则找右子树的最左节点
            TreeLinkNode *p = pNode->right;
            while(p->left) p = p->left;
            return p;
        }
        TreeLinkNode *p = pNode;
        while(p->next) { //没右子树,则找第一个当前节点是父节点左孩子的节点
            if(p->next->left == p) return p->next;
            p = p->next;
        }
        return nullptr; //退到了根节点仍没找到,则返回null
    }
};