leetcode230:Kth Smallest Element in a BST（medium）

在上一章博文中记录了leetcode第378道题（详细内容查看上一章），该题是返回一个矩阵中第k小的元素，与该题类似，这道题是返回一个二叉搜索树中第k小的元素。

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

     思路1：由于刚刚做完第378题，所以最简单的思路便是利用优先队列，遍历一遍二叉树（我用的递归方法进行遍历，中序、前序、后序遍历均可），将每个元素存入优先队列中，然后再对优先队列进行取出操作k次，从而即可返回树中第k小元素。代码如下：

public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        PriorityQueue<Integer> dq = new PriorityQueue<Integer>();
        search(root,list);
        int size = list.size();
        for(int i=0;i<size;i++)
            dq.add(list.get(i));
        int num = k;
        while(--k>0)
            dq.poll();
        return dq.poll();
    }
    private void search(TreeNode root,ArrayList<Integer> list){
        if(root == null)
            return;
        else{
            list.add(root.val);
            search(root.left,list);
            search(root.right,list);
        }
    }
}

虽然AC，但这个算法跑了8ms，因为它不是一个好方法，它遍历了很多没用的元素，完全没有利用 binary search tree的特点，所以我们进行改进！

思路2：思路1是暴力遍历了树上所有的节点，明显没有用到二叉搜索树的特点：左子树上的所有值均小于根节点的值，右子树上的所有值均大于根节点的值。其实我们做了很多冗余操作（算法的优化便是寻找冗余），因为我们只需遍历k个结点就可返回第k小的值，从而减少了n-k次遍历，具体思路是：采用中序遍历，先遍历左子树（因为左子树存放的值小），当遍历到k个结点之后，把当前值记下，不再继续遍历，然后返回该值即可，代码如下（注意观察与上面代码段的差别，仅仅多了if条件判断）：

public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        int[] result = new int[1];
        search(root,list,k,result);
        return result[0];
    }
    private void search(TreeNode root,ArrayList<Integer> list,int k,int[] result){
        if(root == null || list.size() == k)
            return;
        else{
            search(root.left,list,k,result);
            if(list.size() < k){
                list.add(root.val);
                result[0] = root.val;
            }
            search(root.right,list,k,result);
        }
    }
}

经过思路2的优化，算法的运行时间从8ms减少到1ms，感觉爽多了！