LintCode 11 二叉查找树中搜索区间

题目:searchRange

---

要求:

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

样例:

如果有 k1 = 10 和 k2 = 22, 你的程序应该返回 [12, 20, 22].    20   /  \  8   22 / \4   12

算法要求:

无

解题思路:

因为要求的是搜索区间，而不是固定的值，查找跟普通的二叉树查找差不多，只是需要限定范围。

算法如下:

    vector<int> searchRange(TreeNode* root, int k1, int k2) {        vector<int> vec;        if (root == NULL) {            return vec;        }        stack<TreeNode*> s;        TreeNode *p = root;        while (p != NULL || !s.empty()) {            while (p != NULL) {                s.push(p);                if (p->val >= k1) {                    p = p->left;                } else {                    p = NULL;                }            }            if (!s.empty()) {                p =s.top();                s.pop();                if (p->val <= k2 && p->val >=k1) {                    vec.push_back(p->val);                }                if (p->val > k2) {                    p = NULL;                } else {                    p = p->right;                }            }        }    }