OneBug-二叉查找树搜索区间

题目

给定两个值 k1 和 k2（k1 < k2）和一个二叉查找树的根节点。找到树中所有值在 k1 到 k2 范围内的节点。即打印所有 x (k1 <= x <= k2) 其中 x 是二叉查找树的中的节点值。返回所有升序的节点值。

样例 1：

输入：
5
k1 = 6, k2 = 10
输出：
[]

样例 2：
输入：
20
/
8 22
/
4 12
k1 = 10, k2 = 22
输出：
[12,20,22]

代码实现

class Solution {
public:
    /**
     * @param root: The root of the binary search tree.
     * @param k1 and k2: range k1 to k2.
     * @return: Return all keys that k1<=key<=k2 in increasing order.
     */
    vector<int> ans;
    void  dfs(TreeNode* node, int k1, int k2) {
       
        if (node->left && node->val>=k1)		//首先往左边走找最小的且符合条件的数
            dfs(node->left, k1, k2);
         if (node->val >= k1 && node->val <= k2)			//若左子树的都遍历完毕，判断本节点值是否满足条件，若是，加入
            ans.push_back(node->val);
        if (node->right && node->val<=k2)		//再往右走找右子树中符合条件的数
            dfs(node->right, k1, k2);
    }
    vector<int> searchRange(TreeNode* root, int k1, int k2) {
        // write your code here
        if (root==NULL)
            return ans;
        dfs(root, k1, k2);
        return ans;
    }
};

想法：

由于题目要求是要返回升序的节点值，且根据二叉查找树自身的结构特点，用DFS是最佳选择.

来源：CSDN

作者：KeepCodingKeepStudyi

链接：https://blog.csdn.net/hzl90572/article/details/88650585