1.LeetCode 347前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)$ , n 是数组的大小。
- 题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的。
- 你可以按任意顺序返回答案。
思路:
这道题目主要涉及到如下三块内容:
- 要统计元素出现频率
- 对频率排序
- 找出前K个高频元素
首先统计元素出现的频率,这一类的问题可以使用map来进行统计。
然后是对频率进行排序,这里我们可以使用一种 容器适配器就是优先级队列。
1.我们要创建一个map,来统计nums出现的每个次数的次数,第一个代表元素,第二个代表出现的频率。
2.我们需要创建一个优先队列,要根据map中的出现的次数进行排序。
3.使用Map.entry获取map的的字典关系,存入到优先队列中。
4.走K次循环,每次去队列的第一个元素(之前已经进行排序),存入到array数组中,最后返回array数组。
class Solution {
public int[] topKFrequent(int[] nums, int k) {
// 优先队列,详细用法可以百度,稍后晚点补充
PriorityQueue<int[]> pq = new PriorityQueue<>((pair1, pair2)->pair2[1]-pair1[1]);
Map<Integer,Integer> map = new HashMap<>();
for(int num : nums){
map.put(num,map.getOrDefault(num,0)+1);
}
int[] array = new int[k];
for(Map.Entry<Integer,Integer> entry: map.entrySet()){
pq.add(new int[]{entry.getKey(),entry.getValue()});
}
for(int i =0;i < k;i++){
array[i] = pq.poll()[0];
}
return array;
}
}
2.LeetCode 滑动窗口最大值
题目链接:链接
文字讲解及代码:文字讲解及可运行的代码
视频讲解:视频讲解,加深印象
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
提示:
- 1 <= nums.length <= 10^5
- -10^4 <= nums[i] <= 10^4
- 1 <= k <= nums.length
1.我们需要创建一个双向队列
2.我们需要判断,队列头节点在[i-k+1,i]的范围中,不符合的就要弹出
3.我们要对列表的尾部判断,我们放进去的数字要比尾部的大,否则也弹出
4.当i>k-1的时候,第一个窗口也就滑动完了,队列的第一个元素就是第一个滑动窗口的最大值,我们把他取出来放在数组中,最后返回数组
代码如下:
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
// 创建一个双向队列
ArrayDeque<Integer> dq = new ArrayDeque<>();
int n = nums.length;
// 新的返回数组
int[] res = new int[n-k+1];
int index = 0;
for(int i =0;i <n;i++){
// 判断
while (!dq.isEmpty() && dq.peek() < i-k+1){
dq.poll();
}
//2.既然是单调,就要保证每次放进去的数字要比末尾的都大,否则也弹出
while (!dq.isEmpty()&& nums[dq.peekLast()] < nums[i]){
dq.pollLast();
}
dq.offer(i);
// 因为单调,当i增长到符合第一个k范围的时候,每滑动一步都将队列头节点放入结果就行了
if(i >= k-1){
res[index++] = nums[dq.peek()];
}
}
return res;
}
}