【二叉树】【小顶堆】leetcode.230. 二叉搜索树中第K小的元素

GoingJack

于 2021-04-26 22:03:19 发布

阅读量145

点赞数

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/qq_42418668/article/details/116175377

版权

leetcode 专栏收录该内容

37 篇文章

订阅专栏

在这里插入图片描述

/**
 * 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:
    priority_queue<int, vector<int>, less<int>> heap;
    int k_;
    int kthSmallest(TreeNode* root, int k) {    
        k_ = k;
        PreOrder(root);
        return heap.top();
    }
    void PreOrder(TreeNode* root) {
        if(root != nullptr) {
            heap.push(root->val);
            if(heap.size() > k_) {
                heap.pop();
            }
            PreOrder(root->left);
            PreOrder(root->right);
        }
    }
};