给定一颗二叉搜索树,请找出其中的第k大的结点。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按结点数值大小顺序第三个结点的值为4。
#####理解:
想要找出第k大的结点,能够对二叉树的结点val进行排序是最好不过,并且辅助很多类似问题的解答。
对二叉搜索树进行中序遍历后的顺序正是从小到大的排序,所以思路很清楚:首先进行中序遍历,然后根据需要提取出第k个结点。
注意将中序遍历写到一个递归方法中。
import java.util.ArrayList;
/*树结点类定义
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public TreeNode KthNode(TreeNode pRoot, int k)
{
ArrayList<TreeNode> list = new ArrayList<TreeNode>();
if(pRoot == null || k < 1){
return null;
}
KthNodeMid(pRoot, list);//先跑一下中序遍历,得到序列list
if(k > list.size()){
return null;
}
return list.get(k-1);//在直接按序抽取即可
}
private void KthNodeMid(TreeNode pRoot, ArrayList<TreeNode> list){
if(pRoot == null){
return ;
}
KthNodeMid(pRoot.left, list);
list.add(pRoot);
KthNodeMid(pRoot.right, list);
}
}