LeetCode 刷题 [C++] 第230题.二叉搜索树中第K小的元素

题目描述

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

题目分析

由于二叉搜索树的中序遍历为递增序列，即任意子结点都满足：左子结点 < 根节点 < 右子结点。因此，本题可以转化为求中序遍历的第k个节点。

1. 递归的结束条件为当前访问节点为空，以及已经访问了k个节点；
2. 递归遍历时记录当前访问了多少节点；
3. 递归访问到第k个节点时，记录节点的结果ans；

Code

class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        this->target = k;
        dfs(root);
        return this->ans;
    }
private:
    int ans{0}, target{0};
    void dfs(TreeNode* root) {
        if (nullptr == root || !target) return;
        dfs(root->left);
        if (!--target) ans = root->val;
        dfs(root->right);
    }
};