二叉搜索树的第k个结点

最新推荐文章于 2025-09-06 09:26:50 发布

weixin_33778544

最新推荐文章于 2025-09-06 09:26:50 发布

阅读量54

点赞数

CC 4.0 BY-SA版权

文章标签： python java 数据结构与算法

原文链接：https://my.oschina.net/liyurong/blog/1678807

2019独角兽企业重金招聘Python工程师标准>>>

题目描述

给定一颗二叉搜索树，请找出其中的第k大的结点。例如， 5 / \ 3 7 /\ /\ 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

【解决】

① 使用一个变量记录遍历的个数

class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Solution {
int count = 0;
TreeNode KthNode(TreeNode pRoot, int k) {
if (pRoot == null || k <= 0){
return null;
}
TreeNode res = null;
if (pRoot.left != null){
res = KthNode(pRoot.left,k);
}
count ++;
if (res == null){
if (count == k){
res = pRoot;
return res;
}
}
if (res == null && pRoot.right != null){
res = KthNode(pRoot.right,k);
}
return res;
}
}

② 使用一个变量保存中序遍历的结果

import java.util.ArrayList;
import java.util.List;

class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Solution {
List<TreeNode> list = new ArrayList<>();
TreeNode KthNode(TreeNode pRoot, int k) {
if (k == 0) return null;
inorder(pRoot);
if (k > list.size()) return null;
return list.get(k - 1);
}
public void inorder(TreeNode root){
if (root == null) return;
inorder(root.left);
list.add(root);
inorder(root.right);
}
}