二叉树的下一个结点

题目描述

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

 

思路

以上述二叉树为例子，中序排列结果为： 3 2 4 1 6 5 7。

所以3的下一个结点为2，4的下一个结点为1，7的下一个结点为null。

当前结点分为以下情况：

（1）右子树不为空

因为中序遍历，而右子树存在，则当前结点为当前子树的根结点，所以其下一个结点为其右子树最左结点。

（2）右子树为空

a）如果当前结点是它父结点的左子结点，那么它的下一个结点就是它的父结点。

b）如果当前结点是它父结点的右子结点，则可以沿着指向父结点的指针一直向上遍历，直到找到一个是它父结点的左子节点的结点，如果这样的结点存在，则这个结点的父结点就是要找的下一个结点。

 

代码

/*
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 == nullptr) return nullptr;
        TreeLinkNode* nextNode = nullptr;
        if (pNode->right != nullptr){
            auto cur = pNode->right;
            while (cur->left != nullptr) cur = cur->left;
            nextNode = cur;
        }
        else if (pNode->next != nullptr){
            auto p = pNode->next;
            auto cur = pNode;
            while (p && p->left != cur){
                cur = p;
                p = p->next;
            }
            nextNode = p;
        }
        return nextNode;
    }
};