108.Kth Smallest Element in a BST

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note: 
You may assume k is always valid, 1 ≤ k ≤ BST's total elements.

Follow up:
What if the BST is modified (insert/delete operations) often and you need to find the kth smallest frequently? How would you optimize the kthSmallest routine?

分析:即找中序遇到的第k个结点。

   /**
	 * 给定一颗二叉搜索树,请找出其中的第k大的结点。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按结点数值大小顺序第三个结点的值为4。
	 * 中序遇到的第k个结点。 
	 * @date 20160410
	 */
	 public int kthSmallest(TreeNode root, int k) {
		if(root == null){
			return 0;
		}else{
			TreeNode node = root; 
			Stack stack = new Stack();
			stack.push(node);
			/*node表示入栈的元素,popnode表示弹栈的元素*/
			while(!stack.isEmpty()){
				while(node.left != null){//向左走到尽头,把最左分支的结点都入栈。
					node = node.left;
					stack.push(node);			
				}
				TreeNode popnode= (TreeNode) stack.pop();//弹出栈顶元素
				k--;
				if(k == 0){
					return popnode.val;
				}
				/*如果弹栈的元素有右孩子,则让其右孩子入栈,进行下一次循环*/
				if(popnode.right != null){
					node = popnode.right;
					stack.push(node);
				}
			}
		}
		return 0;
        
    }


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值