二叉树的下一个节点——二叉树学习之路（3）

二叉树的下一个节点

题目描述

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

题

                                                 

分析一下找到二叉树当前节点的下一个节点这个问题，按照中序遍历的左-根-右的顺序，总共有三种情况：

观察发现树中包含的指向父节点的指针被命名为next

 

  实现过程就是：先判断有没有有子树if（pNode->right);有然后更新当前节点为右子树：pNode=pNode->right;之后就开始寻找右子树的左子根了（利用循环看是不是有，有就更新pNode）while(pNode->left) pNode=pNode->left;最后返回找到的下一节点return pNode

  从2，3可以看出，3里面也包含2，所以编写的时候将两部分合并编写（分开就通不过调试）:先循环寻找当前节点的父节点while(pNode->next)，建立变量来存当前节点的父节点auto fpNode=pNode->next;判断当前节点是父节点的左子树if(fpNode->left==pNode);是的话就返回下一节点为父节点return fpNode;不是就用父节点更新当前节点pNode=fpNode，一直找下去

1. 当前节点有右子树(1,2,3,4,5)，那么下一节点的输出就是右子树的最左子根（沿着左指针一直指到底，比如上图的2下一节点为10：2——10），如果右子树没有左子根的话就是输出下一节点就是右子树的根。
2. 当前节点没有右子树（8,9,10,11,12），那么就要判断当前节点的父节点，如果当前节点是父节点的左子树（8，10，12），那么根据左根右的顺序，下一节点就是当前节点的父节点。
3. 当前节点没有右子树，而且是父节点的右子树（9,11），那么就要找父节点的父节点了，再判断父节点是不是爷节点的左子根，是就返回下一节点为爷节点，不是就继续找。
4. 如果都不符合（7），那么就返回空return nullptr;

 

/*
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) return nullptr;
        if(pNode->right)
        {
            pNode=pNode->right;
            while(pNode->left) pNode=pNode->left;
            return pNode;
        }
        while(pNode->next)
        {
            auto fpNode=pNode->next;
            if(fpNode->left==pNode) return fpNode;
            pNode=fpNode;
        }
        return nullptr;
    }
};