剑指offer——二叉搜索树的第k个结点（覆盖TreeSet的compare方法）

题目描述

给定一颗二叉搜索树，请找出其中的第k大的结点。例如， 5 / \ 3 7 /\ /\ 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

错误思路：后序遍历，用count记数，等于k时返回。这个做法没有考虑到如下情况。这样后序遍历是不行的。

这里应该是中序遍历。

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

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

    }

}
*/
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;
	}


}

另外一种：

解题思路：

用TreeSet全部加进去一遍。重写compare方法。然后迭代计数。将该类对象作为实际参数传递给TreeSet集合的构造函数。

TreeSet集合排序有两种方式，Comparable和Comparator区别： 
 1：让元素自身具备比较性，需要元素对象实现Comparable接口，覆盖compareTo方法。 
 2：让集合自身具备比较性，需要定义一个实现了Comparator接口的比较器，并覆盖compare方法， 
  并将该类对象作为实际参数传递给TreeSet集合的构造函数。 第二种方式较为灵活。 

参考文献：http://blog.csdn.net/geek_ymv/article/details/38147047

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

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

    }

}
*/
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
public class Solution {
    TreeSet<TreeNode>treeSet = new TreeSet<>(new TreeNodeComparator());
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null||k==0)return null;
		preorder(pRoot);
		Iterator<TreeNode>iterator = treeSet.iterator();
		int count=1;
		while(iterator.hasNext()){
			if(count==k)return iterator.next();
			iterator.next();
			count++;
		}
		return null;
    }
	public void preorder(TreeNode pRoot){
		treeSet.add(pRoot);
		if(pRoot.left!=null)preorder(pRoot.left);
		if(pRoot.right!=null)preorder(pRoot.right);
	}

	
	class TreeNodeComparator implements Comparator<TreeNode>{
		@Override
		public int compare(TreeNode o1, TreeNode o2) {
			return o1.val-o2.val;
		}
	}
	
}