lintcode 11. 二叉查找树中搜索区间

最新推荐文章于 2022-04-02 17:15:38 发布

MissZhou要努力

最新推荐文章于 2022-04-02 17:15:38 发布

阅读量224

点赞数

分类专栏： leetcode 文章标签： leetcode 二叉树

本文链接：https://blog.youkuaiyun.com/zhou_yujia/article/details/101119871

版权

leetcode 专栏收录该内容

89 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找指定区间内所有元素的方法，通过中序遍历或递归策略，实现高效查找k1到k2之间的所有键值，返回按升序排列的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

中序遍历一遍就好啦

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: param root: The root of the binary search tree
     * @param k1: An integer
     * @param k2: An integer
     * @return: return: Return all keys that k1<=key<=k2 in ascending order
     */
    vector<int> searchRange(TreeNode * root, int k1, int k2) {
        // write your code here
        vector<int>ans;
        if(root==NULL) return ans;
        TreeNode* p = root;
        stack<TreeNode*>s;
        while(!s.empty()||p) {
            if(p) {
                s.push(p);
                p = p->left;
            } else {
                p = s.top();
                s.pop();
                if (p->val >= k1 && p->val <= k2) {
                    ans.push_back(p->val);
                }
                p = p->right;
            }
        }
        return ans;
    }
};

其实比较好的递归的时候不满足大小范围的子树就不遍历了。。。

class Solution {
public:
    /**
     * @param root: param root: The root of the binary search tree
     * @param k1: An integer
     * @param k2: An integer
     * @return: return: Return all keys that k1<=key<=k2 in ascending order
     */
    vector<int>ans;
    void dfs(TreeNode* root, int k1, int k2) {
        if (root->left && root->val >= k1)
            dfs(root->left,k1,k2);
        if (root->val>=k1 && root->val<=k2) {
            ans.push_back(root->val);
        }
        if (root->right && root->val <= k2) 
            dfs(root->right, k1, k2);
    }
    vector<int> searchRange(TreeNode * root, int k1, int k2) {
        // write your code here
        ans.clear();
        if(root==NULL) return ans;
        dfs(root, k1, k2);
        return ans;
    }
};