215 - 数组中的第K个最大元素
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
nums.sort() #从小到大排列
return nums[-k]
217 - 存在重复元素
给定一个整数数组,判断是否存在重复元素。
如果存在一值在数组中出现至少两次,函数返回 true 。如果数组中每个元素都不相同,则返回 false 。
class Solution:
def containsDuplicate(self, nums: List[int]) -> bool:
nums.sort()
for i in range(1, len(nums)):
if nums[i] == nums[i-1]:
return True
return False
先排序,再看有没有相邻数的相等即可。
230 - 二叉搜索树中第K小的元素
给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明:你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。
# 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 kthSmallest(self, root: TreeNode, k: int) -> int:
q = []
p = root
counter = 0
while p or q:
while p:
q.append(p)
p = p.left
p = q.pop()
counter += 1
if counter == k:
return p.val
p = p.right
二叉搜索树是一种节点值之间具有一定数量级次序的二叉树,对于树中每个节点:
- 若其左子树存在,则其左子树中每个节点的值都不大于该节点值;
- 若其右子树存在,则其右子树中每个节点的值都不小于该节点值。
若上述二叉树中序遍历,为:0 - 1 - 2 - 3 - 5 - 6 - 7 - 8 - 9
本解法实际上是用队列写dfs中的中序遍历,需要注意的是入队和出队的顺序。