STL 容器解题
1、数组中的第 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 个大的 元素
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int>p(nums.begin(),nums.end());
for(int i = 0;i<k-1;++i){
p.pop();
}
return p.top();
}
方法二: 利用排序库函数
sort(nums.begin(),nums.end());
return nums[nums.size()-k];
lambda 表达式定义一个降序排序
int findKthLargest(vector<int>& nums, int k) {
sort(nums.begin(),nums.end(),[](int a,int b){
return a>b;
});
return nums[k-1];
}
python 函数可以直接利用排序的库函数
# 库函数1
num = sorted(num,reverse=True);
# 库函数2
# nums.sort(reverse=True)
return num[k-1]
2、前 K 个高频元素
给定一个非空的整数数组,返回其中出现频率前 k 高的元素
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
说明:你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数
你的算法的时间复杂度必须优于 O(n*logn) , n 是数组的大小
方法一:C++ 版本
思路:先将数据本身和次数统一,放到一个 unordered_map 序列式容器中,然后把 pair<key,value> 作为一个元素插入到优先级队列中
vector<int> topKFrequent(vector<int>& nums, int k) {
priority_queue<pair<int,int>>p;
vector<int>res;
unordered_map<int,int>mp;
for(auto&e: nums){
mp[e]++;
}
// 将数据导入到优先级队列中
for(auto&e: mp){
// 元素的次数作为键,元素本身的内容作为值
p.push(pair<int,int>(e.second,e.first));
}
// 删除堆第一个元素,因为第一个元素就是最大值
// 删除完之后再进行调整,将次大的值调整到堆顶
while(k--){
res.push_back(p.top().second);
p.pop();
}
return res;
}
方法二:python 版本
思路:把所有的元素组合到字典中,然后按照键值从大到小排序,键值就是某一个元素出现的次数,最后返回前 k 个元素
def topKFrequent(self, nums: List[int], k: int) -> List[int]:
res = []
dic = {
}
for i in range