面试高频算法题补充系列：二叉树的下一个节点

前言

了解更多常考高频算法题可以关注
公众号：一个搬砖的胖子
企业面试题库:https://codetop.cc/
小程序：CodeTop

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

题目描述

给定二叉树其中的一个结点，请找出其中序遍历顺序的下一个结点并且返回。
注意，树中的结点不仅包含左右子结点，而且包含指向父结点的指针。

题目分析

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

节点(设为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
    }
};

参考链接

https://www.nowcoder.com/questionTerminal/9023a0c988684a53960365b889ceaf5e
https://mp.weixin.qq.com/s/ug9KoqbrVFMPBTqX-ZaKbA