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.
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?
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?
Inorder we will first get the min(leftmost node), then 2nd min (parent node of min),
then 3rd min(right node of 2nd min)....
Complexity: O(N) - time O(N) - stack
public class Solution { int count = 0; int result = -1; public int kthSmallest(TreeNode root, int k) { traverse(root, k); return result; } public void traverse(TreeNode root, int k) { if(root == null) return; traverse(root.left, k); count++; if(count == k) result = root.val; traverse(root.right, k); } }
思路2: using iteration to store the path to current min node into a stack.
When we pop node, we check if it has left node and add till the leftmost node all the way to leftmost.
public int kthSmallest(TreeNode root, int k) { Deque<TreeNode> st = new LinkedList<>(); int res = 0; while(root != null){//find the min - leftmostnode st.push(root); root = root.left; } while(!st.isEmpty() && k > 0){ TreeNode cur = st.pop(); res = cur.val; k--; cur = cur.right; while(cur != null){//keep adding till the leftmost node of right node st.push(cur); cur = cur.left; } } return res; }
No comments:
Post a Comment