LeetCode 239 Sliding Window Maximum

Given an array nums, there is a sliding window of size k which is moving from the very left of the array to the very right. You can only see the k numbers in the window. Each time the sliding window moves right by one position.

For example,
Given nums = [1,3,-1,-3,5,3,6,7], and k = 3.

Window position Max
————— —–
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
Therefore, return the max sliding window as [3,3,5,5,6,7].

Note:
You may assume k is always valid, ie: 1 ≤ k ≤ input array’s size for non-empty array.

这道题目,需要使用Java优先级队列来实现(堆)。

AP代码如下:
ps:有一个技巧,可以不自定义compatator():将数组中各元素的负值加入到优先级队列中,然后在返回的时候,将队列中元素的负值加入到结果数组中。

public class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(k==0)
            return nums;
        int len = nums.length;
        int[] rst = new int[len - k + 1];
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<Integer>(k, new Comparator<Integer>() {

            @Override
            public int compare(Integer o1, Integer o2) {
                // TODO Auto-generated method stub
                return o1 < o2 ? 1 : (o1 > o2 ? -1 : 0);
            }

        });
        for (int i = 0; i < k - 1; i++) {
            priorityQueue.offer(nums[i]);
        }
        for (int i = 0; i < len - k + 1; i++) {
            priorityQueue.offer(nums[i + k - 1]);
            rst[i] = priorityQueue.peek();
            priorityQueue.remove(nums[i]);
        }
        return rst;   
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值