285 二叉搜索树中的顺序后继

题目描述:
给你一个二叉搜索树和其中的某一个结点,请你找出该结点在树中顺序后继的节点。

结点 p 的后继是值比 p.val 大的结点中键值最小的结点。

示例 1:
在这里插入图片描述
输入: root = [2,1,3], p = 1
输出: 2
解析: 这里 1 的顺序后继是 2。请注意 p 和返回值都应是 TreeNode 类型。

示例 2:
在这里插入图片描述
输入: root = [5,3,6,2,4,null,null,1], p = 6
输出: null
解析: 因为给出的结点没有顺序后继,所以答案就返回 null 了。

注意:
假如给出的结点在该树中没有顺序后继的话,请返回 null
我们保证树中每个结点的值是唯一的

方法1:
主要思路:
(1)先处理给出的结点可能是最大的结点的情形;
(2)再使用中序遍历,找存在的后继结点,主要是使用临时的结点,保存遍历过程中的前一个结点;
(3)使用另外的结点,保存前一个找到的后继结点,注意提前终止递归的条件;

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void helper(TreeNode* root,TreeNode* p,TreeNode*& pre,TreeNode*& res){
        if(root==NULL||res!=NULL){//终止递归
            return;
        }
        helper(root->left,p,pre,res);
        if(res!=NULL){//提前终止递归的条件
            return;
        }
        //判断前一个结点是不是给出的结点,是的话,保存当前的结点
        if(pre!=NULL){
            if(pre->val==p->val){
                res=root;
                return;
            }
        }
        pre=root;//保存前一个结点
        helper(root->right,p,pre,res);
    }
    TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
        if(root==NULL){
            return NULL;
        }
        TreeNode* tmp_node=root;
        //判断给出的结点是不是最大的结点
        while(tmp_node->right){
            tmp_node=tmp_node->right;    
        }
        if(tmp_node->val==p->val){
            return NULL;
        }
        //辅助结点
        TreeNode* res=NULL;
        TreeNode* pre=NULL;
        helper(root,p,pre,res);
        return res;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值