一、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 ≤ 数组的长度。
思路
- 排序,默认是升序
- 利用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
函数?
思路
- 利用215题的思路,先通过前序遍历把数据取出;
- 取出的数据是升序的,直接索引访问即可;
解题代码(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) 需要额外的空间储存结点值