题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
因为二叉搜索数按照中序遍历就是按照从大到小的顺序,所以我们只需要对它进行中序遍历,然后找到第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)
{
if(pRoot == null || k <= 0){
return null;
}
ArrayList<TreeNode> list = new ArrayList<>();
midSort(pRoot, list);
if(k > list.size()){
return null;
}
return list.get(k - 1);
}
public static void midSort( TreeNode pRoot, ArrayList<TreeNode> list){
if(pRoot == null){
return;
}
midSort(pRoot.left, list);
list.add(pRoot);
midSort(pRoot.right, list);
}
}