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