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

牛客网地址：二叉搜索树的第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;
    }
};