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

来源:互联网 发布:ubuntu防火墙在哪里 编辑:程序博客网 时间:2024/06/14 00:04

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


给定两个值 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;}
0 0
原创粉丝点击