LeetCode #285 - Inorder Successor in BST

LawFile

于 2019-01-29 00:11:55 发布

阅读量470

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/LawFile/article/details/86671707

LeetCode 专栏收录该内容

625 篇文章

订阅专栏

本文探讨了在二叉搜索树中寻找给定节点的中序遍历后继节点的问题。通过迭代实现中序遍历，利用辅助栈记录遍历过程，有效地解决了寻找后继节点的难题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

Given a binary search tree and a node in it, find the in-order successor of that node in the BST.

Note: If the given node has no in-order successor in the tree, return null.

Example 1:

Input: root = [2,1,3], p = 1

/ \

1 3

Output: 2

Example 2:

Input: root = [5,3,6,2,4,null,null,1], p = 6

/ \

3 6

/ \

2 4

Output: null

二叉树迭代实现中序遍历，维护pre和cur两个指针，当pre等于目标节点时，cur等于下一个节点。

class Solution {
public:
    TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
        cur=root;
        pre=NULL;
        while(cur!=NULL||!s.empty())
        {
            while(cur!=NULL)
            {
                s.push(cur);
                cur=cur->left;
            }
            if(s.size()>0)
            {
                cur=s.top();
                if(pre==p) return cur;
                s.pop();
                pre=cur;
                cur=cur->right;
            }
        }
        return NULL;
    }
    
private:
    TreeNode* cur;
    TreeNode* pre;
    stack<TreeNode*> s;    
};