zoukankan      html  css  js  c++  java
  • 腾讯//二叉搜索树中第K小的元素

    给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 个最小的元素。

    说明:
    你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。

    示例 1:

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

    示例 2:

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

    进阶:
    如果二叉搜索树经常被修改(插入/删除操作)并且你需要频繁地查找第 k 小的值,你将如何优化 kthSmallest 函数?

    /**
     * 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 calcTreeSize(TreeNode* root){
            if(root == NULL)
                return 0;
            return 1+calcTreeSize(root->left)+calcTreeSize(root->right);
        }
        int kthSmallest(TreeNode* root, int k) {
            if(root == NULL)
                return 0;
            int leftSize = calcTreeSize(root->left);
            if(k == leftSize + 1)
                return root->val;
            else if(leftSize >= k){
                return kthSmallest(root->left,k);
            }else{
                return kthSmallest(root->right,k-leftSize-1);
            }
        }
    };
  • 相关阅读:
    软件质量属性简述
    浅谈MVC架构
    SOA整理
    程序员的自我修养阅读笔记3
    程序员的自我修养阅读笔记2
    程序员的自我修养阅读笔记1
    架构漫谈阅读笔记3
    架构漫谈阅读笔记2
    Weather APP
    php通过==和!==比较NULL和''结果均为真
  • 原文地址:https://www.cnblogs.com/strawqqhat/p/10602456.html
Copyright © 2011-2022 走看看