#230.二叉搜索树中第k个最小的元素

不羁的心 提交于 2019-11-29 09:51:05

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

知识点:
二叉树中序遍历
思路和代码:
使用二叉树中序遍历的方法,找到第k小的数直接退出

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        temp_stack = []  # 存储栈
        min_v = 0
        while root or temp_stack:
            while root:
                temp_stack.append(root)
                root = root.left
            root = temp_stack.pop()
            min_v = root.val
            k-=1
            if k==0:
                break
            root = root.right
        return min_v
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!