题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。
示例1
输入
{5,3,7,2,4,6,8},3
返回值
{4}
说明
按结点数值大小顺序第三小结点的值为4
首先介绍一下二叉搜索树的定义:
二叉查找树(Binary Search Tree,BST),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
方法一:递归
用递归的方式,不断递归深入根节点的左孩子,直到碰到空节点为止,然后回溯输出当前节点。再以同样的方式递归遍历其右孩子。在此期间,每访问一个节点,我们都对k进行减一操作,直到k为0,说明该节点即为第k个节点。
class Solution {
public:
TreeNode *ans = nullptr;
int count = 0;
void dfs(TreeNode *pRoot, int k){
if(count < k && pRoot->left){
dfs(pRoot->left, k);
}
if(++count == k)
ans = pRoot;
if(count < k && pRoot->right){
dfs(pRoot->right, k);
}
}
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(pRoot == nullptr || k <= 0)
return nullptr;
dfs(pRoot, k);
return ans;
}
};
方法二:非递归
死扣递归很多时候还是有必要的,它不仅是一种优美的思路,简洁的代码,更体现的是对函数不断调入与回溯这一过程的整体把握,基于整个递归程序流程的理解再去写非递归会更简单。递归的过程其实就是函数不断的调入,在计算机中每一个函数都是一个栈帧,函数的调入与完成对应入栈与出栈。
非递归版中序遍历,可以利用栈来模拟递归遍历,首先根入栈,然后令根节点的左孩子不断入栈直到为空,弹出栈顶,令其右孩子入栈,重复以上操作,直到遍历结束或者访问第k个节点为止。
class Solution {
public:
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(!pRoot)
return nullptr;
stack<TreeNode*> res;
TreeNode* p = pRoot;
while(!res.empty() || p)//res是空 and 遍历到空节点
{
while(p)
{
res.push(p);
p = p->left;
}
TreeNode* node = res.top();
res.pop();
if(--k == 0)
return node;
p = node->right;
}
return nullptr;
}
};