(lintcode)第11题二叉查找树中搜索区间

要求：

给定两个值 k1 和 k2（k1 < k2）和一个二叉查找树的根节点。找到树中所有值在 k1 到 k2 范围内的节点。即打印所有x (k1 <= x <= k2) 其中 x 是二叉查找树的中的节点值。返回所有升序的节点值。

样例

如果有 k1 = 10 和 k2 = 22, 你的程序应该返回 [12, 20, 22].

20
/  \
8   22
/ \
4   12

这道题只有一种方法就是递归调用，这里就不多说了，代码如下：

/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;
* public TreeNode left, right;
* public TreeNode(int val) {
* this.val = val;
* this.left = this.right = null;
* }
* }
*/

class Solution {
/*
* @param root: param root: The root of the binary search tree
* @param k1: An integer
* @param k2: An integer
* @return: return: Return all keys that k1<=key<=k2 in ascending order
*/
public static ArrayList arrayList=new ArrayList<>();
public List<Integer> searchRange(TreeNode root, int k1, int k2) {
// write your code here
search(root,k1,k2);
//自定义Comparator对象，自定义排序
Comparator c = new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
// TODO Auto-generated method stub
if((int)o1>(int)o2)
return 1;
//注意！！返回值必须是一对相反数，否则无效。jdk1.7以后就是这样。
// else return 0; //无效
else return -1;
}
};
arrayList.sort(c);
return arrayList;

}
private void search(TreeNode root,int k1,int k2){//递归调用遍历它的子节点
if(root!=null){
if(root.val<=k2&&root.val>=k1){
arrayList.add(root.val);//符合条件则加入集合中
}
search(root.left,k1,k2);//遍历左节点
search(root.right,k1,k2);//遍历右节点
}
}
}