您的位置:首页 > 编程语言 > Java开发

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

2017-01-05 19:41 232 查看

本人电子系,只为一学生。心喜计算机,小编以怡情。

给定两个值 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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  LintCode Java