剑指offer - 62 -- 二叉搜索树的第k个结点

题目描述

给定一棵二叉搜索树，请找出其中的第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;
    }
};