Leetcode—230.二叉搜索树中第K小的元素【中等】

2023每日刷题(六十)

Leetcode—230.二叉搜索树中第K小的元素

Leetcode—230.二叉搜索树中第K小的元素【中等】_第1张图片

实现代码

/**
 * 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) {
        int ans = 0, nodeth = 0;
        function<void(TreeNode*)> dfs = [&](TreeNode*root) {
            if(root == nullptr) {
                return;
            }
            dfs(root->left);
            nodeth++;
            if(nodeth == k) {
                ans=root->val;
                return;
            }
            dfs(root->right);
        };
        dfs(root);
        return ans;
    }
};

运行结果

Leetcode—230.二叉搜索树中第K小的元素【中等】_第2张图片
之后我会持续更新,如果喜欢我的文章,请记得一键三连哦,点赞关注收藏,你的每一个赞每一份关注每一次收藏都将是我前进路上的无限动力 !!!↖(▔▽▔)↗感谢支持!

你可能感兴趣的:(LeetCode刷题,leetcode,算法,职场和发展,dfs,经验分享,c++)