剑指 Offer 54. 二叉搜索树的第k大节点

给定一棵二叉搜索树,请找出其中第k大的节点。

 

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 4
示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 4

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

 

思路:右-中-左遍历得到递减的数

我们可以把遍历的数,放在数组里,然后返回第k个就好了,但是空间复杂度o(n),

如果不使用额外结构,我们可以在遍历的时候,到第k个的时候,直接就把当前的元素赋值给一个全局变量,然后返回即可,不用再接着遍历的:

/**
 * 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:
    int num=0;
    int ans = 0;
    void dfs(TreeNode *root,int k){
        if(root==NULL){
            return ;
        }
        else{
            dfs(root->right,k);
            num++;
            if(num==k){
                ans = root->val;
                return ;
            }
            dfs(root->left,k);
        }
    }
    int kthLargest(TreeNode* root, int k) {
        dfs(root,k);
        return ans;

    }
};

原来树的遍历是这样的,要哪个数据,中序遍历直接就拿出来,呀

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值