lintcode11. 二叉查找树中搜索区间

给定一个二叉查找树和范围[k1, k2]。按照升序返回给定范围内的节点值。

样例
样例 1:

输入:{5},6,10
输出:[]
        5
它将被序列化为 {5}
没有数字介于610之间
样例 2:

输入:{20,8,22,4,12},10,22
输出:[12,20,22]
解释:
        20
       /  \
      8   22
     / \
    4   12
它将被序列化为 {20,8,22,4,12}
[12,20,22]介于1022之间
/**
 * 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> num;
        recursion(root,num,k1,k2);
        return num;
    }
    void recursion(TreeNode*root,vector<int> &num,int k1,int k2)
    {
        if(root==NULL)return;
        recursion(root->left,num,k1,k2);
        if(root->val>=k1&&root->val<=k2)
            num.push_back(root->val);
        recursion(root->right,num,k1,k2);
    }
};
发布了369 篇原创文章 · 获赞 15 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43981315/article/details/104008571