面试题 04.06. 后继者

题

面试题 04.06. 后继者
设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。

如果指定节点没有对应的“下一个”节点，则返回null。

示例 1:

输入: root = [2,1,3], p = 1

2
/
1 3

输出: 2
示例 2:

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

  5
 / \
3   6

/
2 4
/
1

输出: null

解题思路

重点 if (p != root&&p->val<root->val) 要利用二叉搜索树特点 将不需要遍历左子树过滤掉

代码

class Solution
{
    bool isReturn = false;

public:
    TreeNode *inorderSuccessor(TreeNode *root, TreeNode *p)
    {
        if (!root)
        {
            return NULL;
        }
        if (p != root&&p->val<root->val)
        {
            auto l = inorderSuccessor(root->left, p);
            if (l)
            {
                return l;
            }
        }
        if (isReturn)
        {
            return root;
        }
        if (p == root)
        {
            isReturn = true;
        }
        return inorderSuccessor(root->right, p);
    }
};