题目描述

给定一棵二叉搜索树，请找出其中的第 k 小的结点。

你可以假设树和 k 都存在，并且 1≤k≤ 树的总结点数。

数据范围 树中节点数量 [1,500]。

样例

输入：root = [2, 1, 3, null, null, null, null] ，k = 3

    2
   / \
  1   3

输出：3

算法

利用二叉搜索树的中序遍历结果递增的性质，每次经过一个节点将k减1，当k减到零时，这个节点就是要找的

时间复杂度 $O(log n)$

参考文献

y总视频讲解

C++ 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* ans;

    TreeNode* kthNode(TreeNode* root, int k) {
        inorder(root, k);
        return ans;
    }

    void inorder(TreeNode* root, int& k){
        // 二叉搜索树中序遍历递增
        if(!root) return;

        inorder(root->left, k);

        k--;
        if(k == 0) ans = root;

        inorder(root->right, k);
    }
};