描述

給定兩個值 k1 和 k2（k1 < k2）和一個二叉查詢樹的根節點。找到樹中所有值在 k1 到 k2 範圍內的節點。即列印所有x (k1 <= x <= k2) 其中 x 是二叉查詢樹的中的節點值。返回所有升序的節點值。

您在真實的面試中是否遇到過這個題？  是

樣例

如果有 k1 = 10 和 k2 = 22, 你的程式應該返回 [12, 20, 22].

    20
   /  \
  8   22
 / \
4   12

題目連結

程式

/**
 * 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> res;
    int key1, key2;

    void traverse(TreeNode * root){//採用前序遍歷
        if(root == NULL)
            return;
        if(key1 <= root->val && key2 >= root->val)//滿足條件的就存入
            res.push_back(root->val);
        traverse(root->left);
        traverse(root->right);
    }
    vector<int> searchRange(TreeNode * root, int k1, int k2) {
        // write your code here
        if(root == NULL)
            return res;
        key1 = k1;//全部變數值初始化
        key2 = k2;//全部變數值初始化
        traverse(root);
        sort(res.begin(),res.end());//採用升序排序
        return res;
    }
};