牛客网地址:二叉搜索树的第k个结点
题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如(5,3,7,2,4,6,8 )中,
按结点数值大小顺序第三小结点的值为4。
思路
二叉搜索树(Binary Search Tree),(又:二叉查找树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
二叉树的中序遍历首先递归调用遍历左子树,然后访问当前根节点,最后递归调用右子树。
所以可以直接利用二叉搜索树的性质和中序遍历来求解。
代码
//
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
// 递归1
class Solution {
public:
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(!pRoot||k<=0)
return nullptr;
vector<TreeNode*> tmp;
inOrderTraverse(pRoot,tmp);
if(k>tmp.size())
return nullptr;
return tmp[k-1];
}
void inOrderTraverse(TreeNode* head, vector<TreeNode*> &tmp)
{
if(!head)
return;
inOrderTraverse(head->left, tmp);
tmp.emplace_back(head);
inOrderTraverse(head->right, tmp);
}
};
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
// 递归2
class Solution {
public:
int idx = 0;
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(!pRoot||k<=0)
return nullptr;
TreeNode* node = KthNode(pRoot->left, k);
if(node)
return node;
idx += 1;
if(idx == k)
return pRoot;
node = KthNode(pRoot->right, k);
if(node)
return node;
else
return nullptr;
}
};
/*
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||k<=0)
return nullptr;
int idx = 0;
stack<TreeNode*> st;
TreeNode* node = pRoot;
while(node || !st.empty())
{
if(node){
st.push(node);
node = node->left;
}
else{
node = st.top();
st.pop();
idx += 1;
if(idx == k)
return node;
node = node->right;
}
}
return nullptr;
}
};