LeetCode 13 二叉搜索树中第K小的元素

最新推荐文章于 2025-02-10 06:00:00 发布

原创最新推荐文章于 2025-02-10 06:00:00 发布 · 121 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #二叉树中第K小的元素

LeetCode学习专栏收录该内容

27 篇文章

订阅专栏

本文探讨了在二叉搜索树中查找第k个最小元素的问题，提供了两种解决方案：一种使用列表进行中序遍历，另一种采用更优的递归方式，直接在遍历过程中找到目标元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

二叉树中的最大路径和 - LeetCode

问题描述
问题分析
代码

问题描述

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

问题分析

寻找第k小的元素，即现将所有元素用中序遍历排序后输出下标第k-1个元素
后来又发现；列表是没有必要的

代码

#用列表中转中序遍历
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    node_list=[]
    def inorder(self,head):
        if head==None:
            return
        self.inorder(head.left)
        self.node_list.append(head.val)
        self.inorder(head.right)
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        self.node_list=[]
        self.inorder(root)
        return self.node_list[k-1]

#改进版本
class Solution(object):
    def kthSmallest(self, root, k):
        self.count=k
        self.res=0
        def core(root):
            if  root :
                core(root.left)
                self.count=self.count-1
                if self.count==0:
                    self.res=root.val 
                core(root.right)

        core(root)
        return self.res