简单算法 二叉搜索树的第k个结点 (java)
描述
给定一棵二叉搜索树,请找出其中的第k小的TreeNode结点。
示例1
输入:
{5,3,7,2,4,6,8},3
返回值:
4
说明:
按结点数值大小顺序第三小结点的值为4
代码:(中序遍历获取搜索树的顺序,然后直接输出第k个元素)
TreeNode KthNode(TreeNode pRoot, int k) {
if(k == 0) return null;
ArrayList<TreeNode> list = new ArrayList<>();
find(pRoot, list);
if(k > list.size())
return null;
else return list.get(k-1);
}
public void find(TreeNode pRoot , ArrayList<TreeNode> list){
if(pRoot == null) return ;
find(pRoot.left,list);
list.add(pRoot);
find(pRoot.right,list);
}