【DFS】二叉搜索树中第k小的元素

一、题目

. - 力扣（LeetCode）. - 备战技术面试？力扣提供海量技术面试资源，帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/kth-smallest-element-in-a-bst/

二、题解

BST的中序遍历是有序的，可以利用这一性质，对其进行中序遍历的同时维护一个全局的cnt记录遍历过的数量，当到达第k个位置时开始 回溯

/**
 * 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 cnt = 0;

    int dfs(TreeNode* root, int k) {
        if (root == nullptr) return 0;
        int l = dfs(root->left, k);
        if (l) return l;
        
        cnt++;
        if (cnt == k) return root->val;
        
        int r = dfs(root->right, k);
        return r;
    }

    int kthSmallest(TreeNode* root, int k) {
        return dfs(root, k);
    }
};