本人电子系,只为一学生。心喜计算机,小编以怡情。
给定两个值 k1 和 k2(k1 < k2)和一个二叉查找树的根节点。找到树中所有值在 k1 到 k2 范围内的节点。即打印所有x (k1 <= x <= k2) 其中 x 是二叉查找树的中的节点值。返回所有升序的节点值。
思路:二叉树先序遍历非递归便可,十分简单
public ArrayList<Integer> searchRange(TreeNode root, int k1, int k2) {
// write your code here
if(root==null) return new ArrayList<>();
ArrayList<Integer> ret=new ArrayList<>();
TreeNode p=root;
Stack<TreeNode> sta=new Stack<>();
while(p!=null||!sta.isEmpty())
{
if(p!=null)
{
sta.push(p);
p=p.left;
}
else {
p=sta.pop();
if(p.val<=k2&&p.val>=k1)
ret.add(p.val);
p=p.right;
}
}
return ret;
}