ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

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

2019-08-29 19:08:44  阅读:201  来源: 互联网

标签:count python 树中 self digui 二叉 result null root


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

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有