215. 数组中的第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、取倒数第k个数
代码:
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
nums.sort()
return nums[-k]
属于抄近道了,写了冒泡排序,结果超时了
代码:
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
for i in range(len(nums)):
for j in range(len(nums) - i - 1):
if nums[j] > nums[j + 1]:
a, b = nums[j], nums[j + 1]
nums[j], nums[j + 1] = b, a
return nums[-k]
217. 存在重复元素
给定一个整数数组,判断是否存在重复元素。
如果存在一值在数组中出现至少两次,函数返回 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
尝试思路1:
循环判断当前元素是否在后面出现,若有则直接返回true,否则全部判断完返回false
代码:
class Solution:
def containsDuplicate(self, nums: List[int]) -> bool:
for i in range(len(nums)):
if nums[i] in nums[i + 1:]:
return True
return False
【错误】超时
尝试思路2:
排序后判断是否与下一个元素相同
代码:
class Solution:
def containsDuplicate(self, nums: List[int]) -> bool:
nums.sort()
for i in range(len(nums) - 1):
if nums[i] == nums[i + 1]:
return True
return False
230. 二叉搜索树中第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
进阶: 如果二叉搜索树经常被修改(插入/删除操作)并且你需要频繁地查找第 k 小的值,你将如何优化 kthSmallest 函数?
尝试思路:递归
代码:
# 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:
self.l = []
def getl(node):
if node:
self.l.append(node.val)
getl(node.left)
getl(node.right)
getl(root)
self.l.sort()
return self.l[k - 1]