【Leetcode】Kth Smallest Element in a BST

最新推荐文章于 2022-08-12 12:05:29 发布

qbt4juik

最新推荐文章于 2022-08-12 12:05:29 发布

阅读量280

点赞数

分类专栏： Leetcode 文章标签： Leetcode Tree

本文链接：https://blog.csdn.net/qbt4juik/article/details/50604783

版权

Leetcode 专栏收录该内容

112 篇文章 0 订阅

订阅专栏

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?

这道题其实很好地利用BST的特点就好了，举个栗子：

/ \

1 3

我怎么找到第kth的？如果我能排序的话，是不是特别简单。所以我一眼就看出来了，这其实不就是中序遍历放进一个queue就好了么。。。

当然，这里要注意，queue和stack在java里不一样，stack可以直接new，而queue是一个interface，所以new的时候只能 new LinkedList()或者new ArrayDeque();

然后就是巧用queue做中序遍历：

package testAndfun;

import java.util.ArrayDeque;
import java.util.Queue;


public class KthSmallestEle {
	Queue<Integer> queue = new ArrayDeque<Integer>();
    
	
    public int kthSmallest(TreeNode root, int k){
        foo(root);
        while(k>1){
            queue.poll();
            k--;
        }
        return queue.poll();
    }
    
    public void foo(TreeNode root) {
    	 if(root.left!=null){
             foo(root.left);
             queue.add(root.val);
             if(root.right!=null){
                 foo(root.right);
             }
         }
         else{
             queue.add(root.val);
             if(root.right!=null){
                 foo(root.right);
             }
         }
    }
    
    class TreeNode{
    	int val;
    	TreeNode left;
    	TreeNode right;
    	TreeNode(int x){
    		val = x;
    	}
    }
}