给定两个值 k1 和 k2(k1 < k2)和一个二叉查找树的根节点。找到树中所有值在 k1 到 k2 范围内的节点。即打印所有x (k1 <= x <= k2) 其中 x 是二叉查找树的中的节点值。返回所有升序的节点值。
样例
如果有 k1 = 10 和 k2 = 22, 你的程序应该返回 [12, 20, 22].
标签
二叉查找树 二叉树
思路
结合中序遍历,将遍历的节点与阈值点比较,符合条件的节点接入返回数组中。
code
class TreeNode {public: int val; TreeNode *left, *right; TreeNode(int val) { this->val = val; this->left = this->right = NULL; }};class Solution {public: /** * @param root: The root of the binary search tree. * @param k1 and k2: range k1 to k2. * @return: Return all keys that k1<=key<=k2 in ascending order. */ vector searchRange(TreeNode* root, int k1, int k2) { // write your code here if(root == NULL) return vector (); vector result; inorderTraversal(root, result, k1, k2); return result; } void inorderTraversal(TreeNode *root, vector &order, int k1, int k2) { if(root->left != NULL) inorderTraversal(root->left, order, k1, k2); if(root->val >= k1 && root->val <= k2) { order.push_back(root->val); } if(root->right != NULL) inorderTraversal(root->right, order, k1, k2); }};