精选50题之 230. 二叉搜索树中第K小的元素

原题目链接

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

进阶：
如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化 kthSmallest 函数？

直接实现

二叉搜索树已经有序，而对数的最基础的三种搜索算法（前序、中序、后序）中，只有中序搜索可以实现对节点的val按照升序排列逐一访问，因此可以用中序搜索实现。

class Solution {
public:
    void inOrderTraversal(TreeNode *Node, int &tmp, int &val, int k){
        if(!Node)
            return;
        inOrderTraversal(Node -> left, tmp, val, k);
        if(++tmp == k)
        {
            val = Node -> val;
            return;
        }
        inOrderTraversal(Node -> right, tmp, val, k);
    }
    
    int kthSmallest(TreeNode* root, int k) {
        int val = 0, tmp = 0;
        inOrderTraversal(root, tmp, val, k);
        return val;
    }
};

复杂度分析：

• 时间复杂度：O(k)
• 空间复杂度：O(k)

其他方式

不使用递归调用：

class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        stack<TreeNode*> s;
        int num = 0;
        TreeNode *cur = root;
        while(!s.empty() || cur)
        {
            if(cur)
            {
                s.push(cur);
                cur = cur -> left;
            }
            else
            {
                cur = s.top();
                s.pop();
                num ++;
                if(num == k)
                    return cur -> val;
                cur = cur -> right;
            }
        }
        return 0;
    }
};

写在最后

晚安