方法一:二叉搜索树的中序遍历为递增的,所以进行一遍中序遍历将结点放入列表中,然后找到列表的第k个值即可。
/*
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 {
ArrayList<TreeNode> list=new ArrayList<>();
TreeNode KthNode(TreeNode pRoot, int k)
{
if(k<=0){
return null;
}
helper(pRoot);
if(k>list.size()){
return null;
}
return list.get(k-1);
}
void helper(TreeNode pRoot){
if(pRoot==null){
return;
}
helper(pRoot.left);
list.add(pRoot);
helper(pRoot.right);
}
}
方法二:定义一个计数器,计算遍历到了第几个结点。
/*
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 {
int index=0;
TreeNode KthNode(TreeNode pRoot, int k)
{
if(pRoot!=null){
TreeNode node=KthNode(pRoot.left,k);
if(node!=null){
return node;
}
index++;
if(index==k){
return pRoot;
}
node=KthNode(pRoot.right,k);
if(node!=null){
return node;
}
}
return null;
}
}