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

最新推荐文章于 2021-03-16 15:18:34 发布

原创最新推荐文章于 2021-03-16 15:18:34 发布 · 622 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LintCode #Java

LintCode 专栏收录该内容

64 篇文章

订阅专栏

本文介绍了一种在二叉查找树中寻找指定区间内所有节点的方法。通过非递归的先序遍历方式，实现了对给定范围内节点值的有效搜索，并按升序返回结果。

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

给定两个值 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;
}