题目链接:http://www.lintcode.com/zh-cn/problem/search-range-in-binary-search-tree/
1、描述
给定两个值 k1 和 k2(k1 < k2)和一个二叉查找树的根节点。找到树中所有值在 k1 到 k2 范围内的节点。即打印所有x (k1 <= x <= k2) 其中 x 是二叉查找树的中的节点值。返回所有升序的节点值。
2、样例
3、思路分析
要充分利用二叉查找树的特性:左小右大和中序遍历就是一个有序序列。
借助一个栈
想清楚进栈和出站的触发事件即可。
package com.hs; import java.util.ArrayList; import java.util.Stack; // Definition of TreeNode: 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: The root of the binary search tree. * @param k1 and k2: range k1 to k2. * @return: Return all keys that k1<=key<=k2 in ascending order. */ public ArrayList<Integer> searchRange(TreeNode root, int k1, int k2) { //中序遍历 ArrayList<Integer > list=new ArrayList<Integer>(); Stack<TreeNode> S=new Stack<TreeNode>(); if(root!=null){ TreeNode p=root; S.push(p); p=p.left; while(!S.isEmpty()||p!=null){ if(p!=null){ if(p.val>=k1){ S.push(p); p=p.left; }else { p=p.right; } }else { p=S.pop(); //这个if判断使我们的程序通过了测试 if(p.val<k1){//因为头结点没有经过判断就进入栈中,所以需要特殊考虑下,思维要紧密 p=p.right; continue; } if(p.val<=k2){ list.add(new Integer(p.val)); p=p.right; }else { return list; } } } } return list; } }
时间: 2024-10-05 04:24:11