【题目1】
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
思路:单调队列
单调队列
class MyQueue {
Deque<Integer> deque = new LinkedList<>();
//弹出元素时,比较当前要弹出的数值是否等于队列出口的数值,如果相等则弹出
//同时判断队列当前是否为空
void poll(int val) {
if (!deque.isEmpty() && val == deque.peek()) {
deque.poll();
}
}
//添加元素时,如果要添加的元素大于入口处的元素,就将入口元素弹出
//保证队列元素单调递减
//比如此时队列元素3,1,2将要入队,比1大,所以1弹出,此时队列:3,2
void add(int val) {
while (!deque.isEmpty() && val > deque.getLast()) {
deque.removeLast();
}
deque.add(val);
}
//队列队顶元素始终为最大值
int peek() {
return deque.peek();
}
}
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 1) {
return nums;
}
int len = nums.length - k + 1;
//存放结果元素的数组
int[] res = new int[len];
int num = 0;
//自定义队列
MyQueue myQueue = new MyQueue();
//先将前k的元素放入队列
for (int i = 0; i < k; i++) {
myQueue.add(nums[i]);
}
res[num++] = myQueue.peek();
for (int i = k; i < nums.length; i++) {
//滑动窗口移除最前面的元素,移除是判断该元素是否放入队列
myQueue.poll(nums[i - k]);
//滑动窗口加入最后面的元素
myQueue.add(nums[i]);
//记录对应的最大值
res[num++] = myQueue.peek();
}
return res;
}
}
【题目2】优先级队列
给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
这道题目主要涉及到如下三块内容:
- 要统计元素出现频率
- 对频率排序
- 找出前K个高频元素
首先统计元素出现的频率,这一类的问题可以使用map来进行统计。优先级队列其实就是一个披着队列外衣的堆,因为优先级队列对外接口只是从队头取元素,从队尾添加元素,再无其他取元素的方式,看起来就是一个队列。
堆是一棵完全二叉树,树中每个结点的值都不小于(或不大于)其左右孩子的值。 如果父亲结点是大于等于左右孩子就是大顶堆,小于等于左右孩子就是小顶堆。
- 每插入一个节点,先插入最后一个位置然后,依次和父节点进行比较,如果此节点大于父节点,则交换两个节点,直至根节点
- 删除根节点,将最后一个元素放到根节点位置,然后从根节点依次调整,如果根节点大于值比较大的子节点,则跟换他们
class Solution {
public int[] topKFrequent(int[] nums, int k) {
int[] result = new int[k];
HashMap<Integer,Integer> map = new HashMap<>();
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
Set<Map.Entry<Integer,Integer>> entries = map.entrySet();
// 小优先
PriorityQueue<Map.Entry<Integer,Integer>> queue = new PriorityQueue<>(
(o1,o2)->o1.getValue()-o2.getValue()
);
for(Map.Entry<Integer,Integer> entry:entries){
queue.offer(entry);
if(queue.size()>k)
queue.poll();
}
for(int i =k-1;i>=0;i--)
result[i] = queue.poll().getKey();
return result;
}
}
【操作】list转数组操作,流操作
public int[] maxSlidingWindow(int[] nums, int k) {
List<Integer> list = new ArrayList<>();
Queue<Integer> queue = new LinkedList<>();
int i = 0;
while(i<nums.length){
queue.offer(nums[i++]);
if(queue.size()==k){
list.add(queue.stream().max(new Comparator<Integer>(){
public int compare(Integer o1,Integer o2){
return o1-o2;
}
}).get());
queue.poll();
}
}
int[] max = new int[list.size()];
max = list.stream().mapToInt(Integer::valueOf).toArray();
return max;
}