题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
思路
因为是二叉搜索树,所以中序遍历把结点都加到集合中,为从小到大排列的,取第k-1个即可
代码
/*
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类型,所以不可以<Integer>
ArrayList<TreeNode> list = new ArrayList<>();
TreeNode KthNode(TreeNode pRoot, int k)
{
if(pRoot == null) return null;
addNode(pRoot);
if(k >= 1 && list.size() >= k){
return list.get(k-1);//因为索引从0开始
}
return null;
}
//注意中序遍历的写法
public void addNode(TreeNode node){
//国际惯例,先写递归结束的条件
if(node == null) return;
addNode(node.left);
list.add(node);
addNode(node.right);
}
}