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;
                }
            }
        }
    }