[leetcode]python3 演算法攻略-二叉搜尋樹中第K小的元素
阿新 • • 發佈:2018-12-24
給定一個二叉搜尋樹,編寫一個函式 kthSmallest
來查詢其中第 k 個最小的元素。
說明:
你可以假設 k 總是有效的,1 ≤ k ≤ 二叉搜尋樹元素個數。
方案一:中序排序後,返回第K-1個值
class Solution(object): def kthSmallest(self, root, k): """ :type root: TreeNode :type k: int :rtype: int """ def inorderTraversal(root): if root is None: return [] res = [] res.extend(inorderTraversal(root.left)) res.append(root.val) res.extend(inorderTraversal(root.right)) return res return inorderTraversal(root)[k - 1]