编程语言
首页 > 编程语言> > 二叉搜索树中第k小的元素(python实现)

二叉搜索树中第k小的元素(python实现)

作者:互联网

题目描述:
给定一个二叉搜索树,编写一个函数 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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst

思路:搜索二叉树的递归中序遍历
代码:

class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        self.result=0
        self.count=0
        self.digui(root,k)
        return self.result
     
    def digui(self,root,k):
        if root==None or root.val==None:
            return
        self.digui(root.left,k)
        
        self.count+=1
        if self.count==k:
            self.result=root.val
 
        self.digui(root.right,k)

标签:count,python,树中,self,digui,二叉,result,null,root
来源: https://blog.csdn.net/weixin_44512019/article/details/100144541