java第k小元素_leetcode刷题笔记-230. 二叉搜索树中第K小的元素（java实现）

题目描述

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 个最小元素(从 1 开始计数)。

示例 1：

输入：root = [3,1,4,null,2], k = 1

输出：1

示例 2：

输入：root = [5,3,6,2,4,null,null,1], k = 3

输出：3

提示：

树中的节点数为 n 。

1 <= k <= n <= 104

0 <= Node.val <= 104

进阶：如果二叉搜索树经常被修改(插入/删除操作)并且你需要频繁地查找第 k 小的值，你将如何优化算法？

解题思路

二叉搜索树，中序遍历是升序的，所以中序遍历，找到第k个节点就可以了

解题代码

/**

* Definition for a binary tree node.

* public class TreeNode {

* int val;

* TreeNode left;

* TreeNode right;

* TreeNode() {}

* TreeNode(int val) { this.val = val; }

* TreeNode(int val, TreeNode left, TreeNode right) {

* this.val = val;

* this.left = left;

* this.right = right;

* }

* }

*/

class Solution {

public int kthSmallest(TreeNode root, int k) {

traverse(root, k);

return res;

}

public int res = 0;

public int i = 0;

public void traverse(TreeNode root, int k) {

if(root == null) {

return;

}

traverse(root.left, k);

i++;

if(i == k) {

res = root.val;

return;

}

traverse(root.right, k);

}

}