LC230. 二叉搜索树中第K小的元素(C++|中序遍历|优化|面试)

题目描述

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 个最小元素（从 1 开始计数）。

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1
示例 2：

 

 

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

提示：

树中的节点数为 n 。
1 <= k <= n <= 104
0 <= Node.val <= 104
 

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

果果念

这是一道二叉搜索树的题目，本来没有计划写这个的，但是看了一些题解之后，发现没有这种方法，就在此记录。

本题的时间复杂度O(N)，空间复杂度O(1).

主要思想就是利用树的中序遍历LPR是有序的，再用一个标志记录 这是第几小的元素即可。还有一天就面试了，有同学已经拿到满意的offer了，加油哦！

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        //按照先序遍历获得所有元素
        //root是中序遍历
        int num=0;
        bool flag=false;//没有找到
        int ans=-1;
        midTraverse(root,k,num,flag,ans);
        return ans;
    }
    void midTraverse(TreeNode* root,int k,int& num,bool& flag,int& ans){
        if(root->left==nullptr&&root->right==nullptr){
            num++;
            if(num==k){
                flag=true;
                ans=root->val;
            }
            return;
        }
        // cout<<num<<" "<<k<<endl;
        
        if(flag==false&&root->left!=nullptr&&num<k){
            midTraverse(root->left,k,num,flag,ans);
        }

        if(flag==false){
            num++;
            if(num==k){
                flag=true;
                ans=root->val;
                return;
            }
        }

        if(flag==false&&root->right!=nullptr&&num<k){
            midTraverse(root->right,k,num,flag,ans);
        }
    }
};