题目描述:
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
解题思路:
二叉搜索树的中序排序是递增序列,故得到二叉搜索树的中序序列,然后返回第k-1的元素。
代码(java):
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.*;
public class Solution {
TreeNode KthNode(TreeNode pRoot, int k){
if(pRoot==null || k==0)return null;
ArrayList<TreeNode> list=new ArrayList<TreeNode>();
inOrder(list,pRoot);
return list.size()>=k?list.get(k-1):null;
}
public void inOrder(ArrayList<TreeNode> list,TreeNode pRoot){
if (pRoot==null)return;
else{
inOrder(list,pRoot.left);
list.add(pRoot);
inOrder(list,pRoot.right);
}
}
}