# leetcode-230-Kth Smallest Element in a BST

629人阅读 评论(0)

### Kth Smallest Element in a BST

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.

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.
* struct TreeNode {
*     int val;
*     TreeNode *left;
*     TreeNode *right;
*     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
void zhongxu(TreeNode* root,int* a,int& i){
if(!root) return;
zhongxu(root->left,a,i);
a[i++]=root->val;
zhongxu(root->right,a,i);
}
int kthSmallest(TreeNode* root, int k) {
int a[10000];
int i=1;
zhongxu(root,a,i);
return a[k];
}
};

/**
* 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:
void zhongxu(TreeNode* root,int& i,int& ans,int k){
if(!root) return;
zhongxu(root->left,i,ans,k);

if(i++==k) ans=root->val;// i==k 时， 将答案存在来

zhongxu(root->right,i,ans,k);
}
int kthSmallest(TreeNode* root, int k) {
int i=1,ans;
zhongxu(root,i,ans,k);
return ans;
}
};

/**
* 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) {
vector<int>v;
stack<TreeNode*>s;
TreeNode* p;

s.push(root);
while(!s.empty()){
while(p=s.top())  // 向左走到尽头
s.push(p->left);
s.pop();// 空指针出栈
if(!s.empty()){
p=s.top();//取栈顶元素
s.pop();//栈顶元素出栈
v.push_back(p->val);//加到向量vector中
s.push(p->right);
}
}
return v[k-1];
}
};

1
0

* 以上用户言论只代表其个人观点，不代表CSDN网站的观点或立场
个人资料
• 访问：59910次
• 积分：2060
• 等级：
• 排名：第19366名
• 原创：148篇
• 转载：11篇
• 译文：0篇
• 评论：0条
阅读排行