import java.util.*;
public class Solution {
public int KthNode (TreeNode root, int k) {
if (root == null || k == 0){
return - 1;
}
// write code here
List list = new ArrayList();
dfs(root, list);
Collections.sort(list);
return list.size() < k ? -1 : list.get(k - 1);
}
private void dfs (TreeNode root, List list ){
if (root == null){
return;
}
list.add(root.val);
dfs(root.left, list);
dfs(root.right, list);
}
}
官方题解:使用递归,中序遍历
import java.util.*;
public class Solution {
// 题解
// 首先他是二叉搜索树,所以他是有序的
// 设置全局变量记录他的值
int count = 0, res = -1;
public int KthNode (TreeNode root, int k) {
// write code here
dfs(root, k);
return res;
}
private void dfs (TreeNode root, int k){
if (root == null || count >= k){
return;
}
dfs(root.left, k);
count++;
if (count == k){
res = root.val;
}
dfs(root.right, k);
}
}
暑期编程PK赛
得CSDN机械键盘等精美礼品!


