关闭

LeetCode:Kth Smallest Element in a BST

标签: LeetCodeKth Smallest Element
652人阅读 评论(1) 收藏 举报
分类:

Kth Smallest Element in a BST




Total Accepted: 49142 Total Submissions: 128501 Difficulty: Medium

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note: 
You may assume k is always valid, 1 ≤ k ≤ BST's total elements.

Follow up:
What if the BST is modified (insert/delete operations) often and you need to find the kth smallest frequently? How would you optimize the kthSmallest routine?

Hint:

  1. Try to utilize the property of a BST.
  2. What if you could modify the BST node's structure?
  3. The optimal runtime complexity is O(height of BST).

Credits:
Special thanks to @ts for adding this problem and creating all test cases.

Subscribe to see which companies asked this question

Hide Tags
 Binary Search Tree
Hide Similar Problems
 (M) Binary Tree Inorder Traversal





























java code:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        
        int count = countNode(root.left);
        
        if(k == count + 1)
            return root.val;
        else if(k < count + 1)
            return kthSmallest(root.left, k);
        else // k > count + 1
            return kthSmallest(root.right, k - count - 1);
            
    }
    
    // 自定义函数:计算树中结点个数
    int countNode(TreeNode root) {
        if(root == null) return 0;
        
        return countNode(root.left) + countNode(root.right) + 1;
    }
    
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:339474次
    • 积分:6282
    • 等级:
    • 排名:第4060名
    • 原创:312篇
    • 转载:1篇
    • 译文:0篇
    • 评论:23条
    博客专栏
    最新评论