力扣刷题(python)50天——第四十二天：二叉搜索树中第K小的元素

力扣刷题(python)50天——第四十二天：二叉搜索树中第K小的元素

题目描述

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法

递归获取每个节点，放入列表中排序查找。

解答

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

class Solution(object):
    list1=[]
    def kthSmallest(self, root, k):
        """
        :type root: TreeNode
        :type k: int
        :rtype: int
        """
        self.list1+=[root.val]
        if root.left:
            self.find(root.left)
        if root.right:
            self.find(root.right)
        self.list1=list(set(self.list1))
        self.list1.sort()
        print(self.list1)
        return self.list1[k-1]
    def find(self,root):
        self.list1+=[root.val]
        if root.left:
            self.find(root.left)
        if root.right:
            self.find(root.right)

执行结果

提升：

注意本题中提及到的，该二叉树为二叉搜索树，二叉搜索树具有特性：

若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值； 它的左、右子树也分别为二叉排序树

另外可以用中序排列，栈，以及迭代器yield做