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

最新推荐文章于 2024-07-28 16:14:34 发布

Missbubu

最新推荐文章于 2024-07-28 16:14:34 发布

阅读量276

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode

本文链接：https://blog.youkuaiyun.com/Missbubu/article/details/77977382

lintcode 专栏收录该内容

218 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找指定区间内所有元素的方法，通过迭代而非递归的方式实现，确保了返回结果的升序排列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2017.9.14

二叉搜索树的前序遍历就是有序数列。

/**
 * 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;
 *     }
 * }
 */


public 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 List<Integer> searchRange(TreeNode root, int k1, int k2) {
	        // write your code here
		 LinkedList<Integer> res = new LinkedList<Integer>();
		 if(root == null){
			 return res;
		 }
		 // 其实也就是二叉树的前序遍历。
		 LinkedList<TreeNode> stack = new LinkedList<TreeNode>();
		 TreeNode bt = root;
		 while(bt != null || !stack.isEmpty()){
			 while(bt !=null){
				 stack.push(bt);
				 bt = bt.left;
			 }
			 bt = stack.pop();
			 if(bt.val > k2){
				 return res;
			 }
			 if(bt.val >= k1){
				 res.add(bt.val);
			 }
			 bt = bt.right;
		 }
		 return res;
	 }
}