二叉树的迭代排序：230. 二叉搜索树中第K小的元素

最新推荐文章于 2022-05-19 22:15:58 发布

BlackMan_阿伟

最新推荐文章于 2022-05-19 22:15:58 发布

阅读量104

点赞数

分类专栏： Leetcode 树文章标签：二叉树 leetcode 算法数据结构 python

本文链接：https://blog.csdn.net/weixin_37724529/article/details/108073382

版权

Leetcode 同时被 2 个专栏收录

319 篇文章 0 订阅

订阅专栏

树

46 篇文章 0 订阅

订阅专栏

题目链接：230.二叉搜索树的第K小的元素

两个做法：一个是递归，一个是迭代；

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        if not root:
            return []

        #迭代
        stack,res = [],[]
        cur = root
        i=1
        while stack or cur:
            while cur:
                stack.append(cur)
                cur = cur.left
            cur = stack.pop()
            if k==i:
                return cur.val
            cur = cur.right
            i+=1





        #递归
        res = []
        def helper(root):
            if not root:
                return
            helper(root.left)
            res.append(root.val)
            helper(root.right)
        helper(root)

        return res[k-1]