牛客-JZ 57二叉树的下一个节点【二叉树的中序遍历】

问题描述

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

解题报告

首先判断该节点 t 是否有右子树：

• 如果没有的话，我们需要返回到该节点的父节点 p ，检查节点 p 的右孩子是否为节点 t，如果是，回溯法节点 p，继续同样的操作，如果不是，返回当前节点的父节点。
• 如果有的话，那么该节点的右子树的最左边的节点即为节点 t 的下一个节点。

实现代码

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

参考资料

[1] 牛客-JZ 57二叉树的下一个节点