一、No0215. 数组中的第K个最大元素

题目

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例1

• 输入: [3,2,1,5,6,4] 和 k = 2
• 输出: 5

示例2

• 输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
• 输出: 4

说明

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

思路

1. 排序，默认是升序
2. 利用Python索引的特性返回

解题代码(Python3)

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        nums.sort()
        return nums[-k]

复杂度分析：

• 时间复杂度O(n)
• 空间复杂度O(1)

运行结果：

二、No0217. 存在重复元素

题目

给定一个整数数组，判断是否存在重复元素。

如果存在一值在数组中出现至少两次，函数返回 true 。如果数组中每个元素都不相同，则返回 false 。

示例1

• 输入: [1,2,3,1]
• 输出: true

示例2

• 输入: [1,2,3,4]
• 输出: false

示例3

• 输入: [1,1,1,3,3,4,3,2,4,2]
• 输出: true

思路：

• 利用set和list比较长度

解题代码(Python3)

class Solution:
    def containsDuplicate(self, nums: List[int]) -> bool:
        return len(nums) != len(set(nums))

复杂度分析：

• 时间复杂度O(n)
• 空间复杂度O(n) 需要保存set(nums)

运行结果：

三、No0230. 二叉搜索树中第K小的元素

题目

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

说明

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

示例1

• 输入: root = [3,1,4,null,2], k = 1
• 输出: 1

示例2

• 输入: root = [5,3,6,2,4,null,null,1], k = 3
• 输出: 3

进阶

如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化 kthSmallest 函数？

扫描二维码关注公众号，回复： 12454575 查看本文章

思路

1. 利用215题的思路，先通过前序遍历把数据取出；
2. 取出的数据是升序的，直接索引访问即可；

解题代码(Python3)

# 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 __init__(self):
        self.nums = []
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        #前序遍历
        def preOrder(node):
            if node:
                preOrder(node.left)
                self.nums.append(node.val)
                preOrder(node.right)
        preOrder(root)
        return self.nums[k-1]

复杂度分析：

• 时间复杂度 O(n)
• 空间复杂度 O(n) 需要额外的空间储存结点值

运行结果