剑指Offer——二叉搜索树中第K大的节点

题目描述：给定一颗二叉树，找出第K大的节点，这个第K大的节点是从小到大的第K个节点，所以也可以说是第K小的节点。对于一颗二叉树来说，其中序遍历就是节点值从小到大的排列，那么我们用一个ArrayList把中序遍历的结果保存下来，找到第K-1个节点就是第K大的节点。因此很容易想到以下的代码：

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.ArrayList;
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        //中序遍历的第k-1个节点就是第k大的节点
        if(pRoot == null || k<=0)
            return null;
        ArrayList<TreeNode> aList = new ArrayList();
        inOrderRecursive(pRoot,aList);
        int len = aList.size();
        if(len<k)
            return null;
        else
            return aList.get(k-1);
    }
    
    void inOrderRecursive(TreeNode root,ArrayList<TreeNode> al){
        if(root == null){
            return;
        }
        
        inOrderRecursive(root.left,al);
        al.add(root);
        inOrderRecursive(root.right,al);
    }

}

当然，也可以边遍历，边返回节点，这样可以提高效率，比如当这颗二叉树很大，但是K很小的时候，就会大大的节约时间。代码实现如下：

//思路：二叉搜索树按照中序遍历的顺序打印出来正好就是排序好的顺序。
//     所以，按照中序遍历顺序找到第k个结点就是结果。
public class Solution {
   int index = 0; //计数器
    TreeNode KthNode(TreeNode root, int k)
    {
        if(root != null){ //中序遍历寻找第k个
            TreeNode node = KthNode(root.left,k);
            if(node != null)
                return node;
            index ++;
            if(index == k)
                return root;
            node = KthNode(root.right,k);
            if(node != null)
                return node;
        }
        return null;
    }
}