leetcode---腾讯 搜索二叉树中第K小的元素

原题：
给定一个二叉搜索树，编写一个函数 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析：
由于搜索二叉树的中序遍历是升序排列的，所以先将二叉树做中序排列，之后返回第K-1个元素即可

代码：

class TreeNode(object):
    def __init__(self, x):
       self.val = x
       self.left = None
       self.right = None
List = []
def PreorderTraversal(root) :
    #List = []  注意不能在这将List赋值为[]，因为每次递归后都会将其赋空，最后返回的不是想要的数据
    if root==None:
        return
    PreorderTraversal(root.left)
    print(root.val)
    List.append(root.val)
    PreorderTraversal(root.right)
   # return List
def kthSmallest(self, root, k):
        """
        :type root: TreeNode
        :type k: int
        :rtype: int
        """
        return(List[k-1])

root = TreeNode(3)
root.left = TreeNode(1)
root.right = TreeNode(4)
root.left.right = TreeNode(2)
print("二叉树的中序遍历：")
list=PreorderTraversal(root)
self = []
print("第k小的元素：")
print(kthSmallest(self,root,1))

效果：