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?
* Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ publicclass{ publicintkthSmallest(TreeNode root, int k){
int ans = 0; TreeNode knode = root;
int c = count(knode.left); if (c + 1 == k) return knode.val;
if (c < k) return kthSmallest(knode.right, k - 1 - c);// 往右的差值 if (c + 1 > k) return kthSmallest(knode.left, k);
近期评论