LeetCode 230: Kth Smallest Element in a BST 查找二叉排序树

最新推荐文章于 2020-06-21 11:39:02 发布

SunliyMonkey

最新推荐文章于 2020-06-21 11:39:02 发布

阅读量1.8k

点赞数

分类专栏：一一『 LeetCode 』文章标签： leetcode 二叉排序树递归

本文链接：https://blog.csdn.net/SunliyMonkey/article/details/48179133

版权

一一『 LeetCode 』专栏收录该内容

7 篇文章 0 订阅

订阅专栏

博客转载请注明地址：http://blog.csdn.net/SunliyMonkey/article/details/48179133

题目描述

题目地址：https://leetcode.com/problems/kth-smallest-element-in-a-bst/
在一颗二叉排序树当中，寻找第k大的数。

考察点

递归
树的遍历

陷阱

无

Code

/**
 * 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:
    int ret;
    int aim;
    void search(TreeNode* x, int &ord){
        if(!x || ord >= aim)
            return;
        search(x->left, ord);

        ord++;
        if(ord == aim){
            ret = x->val;
            return;
        }

        search(x->right, ord);
    }
    int kthSmallest(TreeNode* root, int k) {
        int ord = 0;
        aim = k;
        search(root, ord);
        return ret;
    }
};