【栈与队列】力扣239. 滑动窗口最大值&347.前 K 个高频元素

最新推荐文章于 2024-11-15 16:27:37 发布

gxzgxzgxz_

最新推荐文章于 2024-11-15 16:27:37 发布

阅读量168

点赞数

文章标签： leetcode 算法职场和发展哈希算法 java

本文链接：https://blog.csdn.net/gxzgxzgxz_/article/details/132761029

版权

本文介绍了如何使用Java编程语言中的数据结构如Deque、PriorityQueue和HashMap来解决两个问题：1)滑动窗口内的最大值；2)找出数组中前K个高频元素。作者通过MyQueue和小顶堆PriorityQueue的数据结构优化了空间复杂度。

摘要由CSDN通过智能技术生成

239. 滑动窗口最大值

题目：

class MyQueue{
    Deque<Integer> deque=new LinkedList<>();

    void poll(int value){
        if (!deque.isEmpty()&&value==deque.peek()){
            deque.pop();
        }
    }

    void add(int value){
        while (!deque.isEmpty()&&value>deque.getLast()){
            deque.removeLast();
        }
        deque.add(value);
    }

    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();

        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;
    }
}

347.前 K 个高频元素

题目：

基于小顶堆实现。

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);
        int[] res = new int[k]; // 答案数组为 k 个元素
        Map<Integer, Integer> map = new HashMap<>(); // 记录元素出现次数
        for(int num : nums) map.put(num, map.getOrDefault(num, 0) + 1);
        for(var x : map.entrySet()) { // entrySet 获取 k-v Set 集合
            // 将 kv 转化成数组
            int[] tmp = new int[2];
            tmp[0] = x.getKey();
            tmp[1] = x.getValue();
            pq.offer(tmp);
            if(pq.size() > k) {
                pq.poll();
            }
        }
        for(int i = 0; i < k; i ++) {
            res[i] = pq.poll()[0]; // 获取优先队列里的元素
        }
        return res;
    }
}