二叉搜索树的第k个结点
1.题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
2.思路
二叉搜索树的中序遍历是递增序列,遍历i记录第几个节点,find记录是否找到,如果i == k则提前结束遍历。
3.代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(pRoot == NULL){
return NULL;
}
int i = 0;
bool find = false;
TreeNode* res = NULL;
findKNode(pRoot,k,i,find,res);
return res;
}
void findKNode(TreeNode* pRoot, int& k, int& i, bool& find, TreeNode*& res){
if(pRoot == NULL || find){
return ;
}
findKNode(pRoot->left,k,i,find,res);
if(++i == k){
res = pRoot;
find = true;
return ;
}
findKNode(pRoot->right,k,i,find,res);
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)