239. 滑动窗口最大值
题目链接:239. 滑动窗口最大值
文档讲解:代码随想录(programmercarl.com)
视频讲解:单调队列正式登场!| LeetCode:239. 滑动窗口最大值
状态:已解决
方法:单调队列
设置一个双端队列,其中存储的元素为数组的下标,使用LinkedList实现。该双端队列的First端一定为该队列中的最大值,具体实现方式为:获取数组当前元素的下标,若队列为空则将该下标值从 Last 端入队,若不为空,则一直与 Last 端的数字代表的数组中的元素比较大小,若大于,则将原队列中的Last端的数从右侧出队,直到队列为空或者小于队列中的元素代表的数组值。
从第 nums[k] 开始,若当前元素与 First 端元素相差大于等于 k,则将 First 端元素从左侧出队。队列中的元素个数一定小于等于 k 个。
Java代码如下:
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
Deque<Integer> d = new LinkedList<Integer>();
for(int i=0;i<k;i++){
while(!d.isEmpty()&&nums[i]>=nums[d.peekLast()]){
d.pollLast();
}
d.offerLast(i);
}
int[] res = new int[nums.length-k+1];
res[0]=nums[d.peekFirst()];
for(int i=0;i<nums.length-k;i++){
if(i>=d.peekFirst()){
d.pollFirst();
}
while(!d.isEmpty()&&nums[i+k]>=nums[d.peekLast()]){
d.pollLast();
}
d.offerLast(i+k);
res[i+1]=nums[d.peekFirst()];
}
return res;
}
}
复杂度分析:
时间复杂度:O(n)
空间复杂度:O(k)
347. 前 K 个高频元素
题目链接:347. 前 K 个高频元素
文档讲解:代码随想录(programmercarl.com)
视频讲解:优先级队列正式登场!大顶堆、小顶堆该怎么用?| LeetCode:347.前 K 个高频元素
状态:已解决
方法:优先级队列
首先使用 HashMap 记录数组中每个元素及其对应的出现次数,对 HashMap 中的元素按照 Value ,即次数进行排序。使用优先级队列(堆)进行排序。
设置优先级队列(堆),该题中更适合使用小顶堆进行解决。保持堆中的元素为 k 个。遍历HashMap,将每个元素添加到堆中,若元素总个数大于 k ,则将堆顶(最小值)弹出。最后堆中保存的元素即为出现次数最多的 k 个。
Java代码如下:
class Solution {
public int[] topKFrequent(int[] nums, int k) {
Map<Integer,Integer> m = new HashMap<Integer,Integer>();
for(int i=0;i<nums.length;i++){
m.put(nums[i],m.getOrDefault(nums[i],0)+1);
}
int[] x = new int[2];
PriorityQueue<int[]> pq = new PriorityQueue<>((o1,o2) -> (o1[1])-o2[1]);
for(Map.Entry<Integer,Integer> ent:m.entrySet()){
x = new int[2];
x[0]=ent.getKey();
x[1]=ent.getValue();
pq.offer(x);
if(pq.size()>k){
pq.poll();
}
}
int[] res = new int[k];
for(int i=k-1;i>=0;i--){
res[i]=pq.poll()[0];
}
return res;
}
}
复杂度分析:
时间复杂度:O(n logk)
空间复杂度:O(N)