STL 容器类型的OJ题篇1

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*logⁿ) , 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