题目
给定一个二叉搜索树的根节点 root
,和一个整数 k
,请你设计一个算法查找其中第 k
个最小元素(从 1 开始计数)。
示例 1:
输入:root = [3,1,4,null,2], k = 1 输出:1
示例 2:文章来源:https://www.toymoban.com/news/detail-694063.html
文章来源地址https://www.toymoban.com/news/detail-694063.html
输入:root = [5,3,6,2,4,null,null,1], k = 3 输出:3
题解
/**
* 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 {
private int ans;// 接收答案
private int p; // 接收k
public int kthSmallest(TreeNode root, int k) {
p = k;
dfs(root);
return ans;
}
private void dfs(TreeNode root) {
//中序遍历
if (root == null) {
return;
}
dfs(root.left);
if (--p == 0) {
ans = root.val;// 更新答案
}
dfs(root.right);
}
}
到了这里,关于每日一题 230二叉搜索树中第K小的元素(中序遍历)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!