LCR 174. 寻找二叉搜索树中的目标节点 (二叉搜索树中序遍历+回溯)

文章介绍了如何在给定的二叉搜索树中,利用中序遍历的逆序特性,通过递归找到第cnt大的员工编号。方法是先递归到最右结点,然后根据cnt减一并判断是否达到目标,直到找到或cnt为0时返回结果。
摘要由CSDN通过智能技术生成

某公司组织架构以二叉搜索树形式记录,节点值为处于该职位的员工编号。请返回第 cnt 大的员工编号。

示例 1:

输入:root = [7, 3, 9, 1, 5], cnt = 2
       7
      / \
     3   9
    / \
   1   5
输出:7

示例 2:

输入: root = [10, 5, 15, 2, 7, null, 20, 1, null, 6, 8], cnt = 4
       10
      / \
     5   15
    / \    \
   2   7    20
  /   / \ 
 1   6   8
输出: 8

提示:

  • 1 ≤ cnt ≤ 二叉搜索树元素个数

思路:

1、二叉搜索树的中序遍历(左根右)为递增序列,我们利用(右根左)即可得到递减序列。先递归至最右结点即最大结点处,利用cnt--是否为0判断当前结点值大小次序是否符合要求,不符合则继续递归当前结点左子树。

2、递归:

终止条件结点为空或已经找到目标结点
递归右子树一直到最大结点处
条件判断

1、cnt--

2、判断cnt是否为0,是则赋值res,直接返回

3、不是则继续递归左子树

返回值返回res

代码实现:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int findTargetNode(TreeNode* root, int cnt) {
        findNode(root, cnt);
        return res;
    }
    void findNode(TreeNode* root, int& cnt)             //cnt需要引用!!!
    {
        if(!root || res != 0)                           //结点为空或者res已被赋值即已经找到目标值
            return;
        findNode(root->right, cnt);                     //递归至最大结点处
        --cnt;
        if(cnt == 0)
        {
            res = root->val;
            return;
        }
        findNode(root->left, cnt);
    }
private:
    int res = 0;
};
  • 9
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值