首页 > 学院 > 开发设计 > 正文

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

2019-11-08 18:38:37
字体:
来源:转载
供稿:网友

题目: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; } } } }
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表