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

最新推荐文章于 2021-10-17 23:59:39 发布

ippputeeel

最新推荐文章于 2021-10-17 23:59:39 发布

阅读量134

点赞数

本文链接：https://blog.csdn.net/weixin_40485472/article/details/81990489

版权

题目：

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

解法一：

利用中序遍历，不需要全部遍历完，找到第K个就可以

/**
 * 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 kthSmallest(TreeNode* root, int k) {
        int travel = 0;
        int kth;
        stack<TreeNode*> s;
        s.push(root);
        TreeNode* last_pop = root;
        while(travel<k){
            TreeNode* p = s.top();
            if(p->left!=nullptr && p->left!=last_pop && p->right!=last_pop){
                s.push(p->left);
            }else if(p->right != nullptr && p->right != last_pop){
                s.push(p->right);
                travel ++;
                kth = p->val;
            }else{
                s.pop();
                last_pop = p;
                if(p->right == nullptr){
                    travel ++;
                    kth = p->val;
                }
            }
        }
        return kth;
    }
};

解法二：先看左子树的节点数left，如果left+1 == k，说明根节点是kth，如果left>=k，则在左子树找最大点，如果left +1<k,则在右子树中找k-left-1（与数组中用二分法找第k小是一样的思路）

/**
 * 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 kthSmallest(TreeNode* root, int k) {
        if(root==NULL) return 0;
        int leftsize = calnode(root->left);
        if(leftsize + 1 == k){
            return root->val;
        }else if(leftsize>=k){
            return kthSmallest(root->left, k);
        }else{
            return kthSmallest(root->right, k-1-leftsize);
        }
    }
    
    int calnode(TreeNode* root){
        if(root==NULL) return 0;
        return 1 + calnode(root->left) + calnode(root->right);
    }
};