每日一题 230二叉搜索树中第K小的元素(中序遍历)
题目
给定一个二叉搜索树的根节点 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
题解
/*** 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; // 接收kpublic 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);}
}