ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

leetcode230. Kth Smallest Element in a BST

2021-09-03 20:58:14  阅读:199  来源: 互联网

标签:right TreeNode BST nullptr int Smallest Kth root left


题目:题目链接

简单概括:求BST(平衡二叉树)的第k小值。

怎么做呢?我感觉就是可以按中序遍历整个二叉树,保存到数组中,然后从数组拿就完了。这个就是O(n)的时间复杂度了。代码也不复杂,就是中序遍历。贴代码:

class Solution {
public:
    vector<int>nums;
    void inOrder(TreeNode* root){
        if(root == nullptr) return;
        inOrder(root->left);
        nums.push_back(root->val);
        inOrder(root->right);
        return;
    }
    int kthSmallest(TreeNode* root, int k) {
        inOrder(root);
        return nums[k-1];
    }
};

还有种做法就是根据BST的性质,左子树的值<父节点的值<右子树的值。 按照我们朴素的思想就是先从左下角的结点开始,然后找中间,然后再找右边的,再循环。他其实就是一个往上搜索的过程。说白了也是中序遍历。只不过是迭代的中序遍历,要用到栈。贴代码吧:

/**
 * 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 kthSmallest(TreeNode* root, int k) {
        stack<TreeNode*> st;
        while (root != nullptr || !st.empty()) {
            while (root != nullptr) {   /*找到最下角的结点,也就是最小的结点*/
                st.push(root);/*把他的父结点都放进去,后面要用到的*/
                root = root->left;
            }
            root = st.top();/*取出当前值最小的结点,这是根据我们上一个while决定的。*/
            st.pop();/*看完了就弹出*/
            if (k-- == 1) return root->val;
            root = root->right;/*最后搜索右边*/
        }/*其实栈的顶部都是当前最小值,只需要根据k一个个弹出直至正确答案就可以了。*/
        return 0;
    }
};

加油啊!!!

标签:right,TreeNode,BST,nullptr,int,Smallest,Kth,root,left
来源: https://blog.csdn.net/weixin_43216252/article/details/120089747

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有