Leetcode Day12——二叉搜索树中第K小的元素

1.Description

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数

来源：力扣（LeetCode）
 

2.Example

3.Code

class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        result = list()
        stack = [root]
        while len(stack):
            top = stack.pop()
            if top.right:
                stack.append(top.right)

            if top.left:
                stack.append(top.left)

            result.append(top.val)

        result.sort()
        return result[k-1]

4.Result