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:
- Try to utilize the property of a BST.
- What if you could modify the BST node's structure?
- The optimal runtime complexity is O(height of BST).
Credits:
Special thanks to @ts for adding this problem and creating all test cases.
/**
* 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 kthSmallest(TreeNode* root, int k) {
stack<TreeNode*> record;
TreeNode* cur = root;
while(cur != nullptr){
record.push(cur);
cur = cur -> left;
}
while(!record.empty()){
TreeNode* curNode = record.top();
record.pop();
k--;
if(k == 0)
return curNode -> val;
TreeNode* cur = curNode -> right;
while(cur != nullptr){
record.push(cur);
cur = cur -> left;
}
}
return -1;
}
};