剑指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;
}
}

}