码迷,mamicode.com
首页 > Windows程序 > 详细

[leetcode]239. Sliding Window Maximum滑动窗口最大值

时间:2018-06-16 10:35:22      阅读:212      评论:0      收藏:0      [点我收藏+]

标签:and   array   取出   you   for   数据   out   span   one   

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. Return the max sliding window.

Example:

Input: nums = [1,3,-1,-3,5,3,6,7], and k = 3
Output: [3,3,5,5,6,7] 
Explanation: 

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

 

 

题意:

给定一个长度为k的滑动窗口不断从左往右滑动,给出过程中的各个最大值。

 

思路:

使用一个每次能取出极值的数据结构,TreeMap,如下图,其底层用BST来储存

技术分享图片

 

TreeMap要求key必须是比较大小(自然排序或定制排序)

以[1,1,-1,-3,5,3,6,7], k = 3 为例, 遍历数组,将数组每个元素作为TreeMap的key, 将该元素出现频率作为对应value

[1,  1,   -1,   -3,  5,  3,  6,  7] 

 ^ i = 0

技术分享图片

[1,  1,   -1,   -3,  5,  3,  6,  7] 

         ^ i = 1

技术分享图片

 

[1,  1,   -1,   -3,  5,  3,  6,  7] 

                   ^   i = 2

技术分享图片

 

[1,  1,   -1,   -3,  5,  3,  6,  7] 

                              ^  i = 3 此时 i >= k  则先将a[i-k]在TreeMap中对应的出现频率(value) 减1 

                                                         再check一下 a[i-k]对应的value是否为0,为0则直接删去。

技术分享图片

 

此例中,a[i-k] = 1, 在TreeMap中对应的value为2,那么value减1 后为1, 仍然继续保留。

技术分享图片

 

由此可以看出,大体思路是用TreeMap维护一个所有value值相加为K的BST

用lastKey()来取出当前TreeMap里最大值(根据BST性质,最大值一定在最右)

 

代码:

 1 class Solution {
 2     public int[] maxSlidingWindow(int[] a, int k) {
 3         // corner case
 4         if(k <= 0) return new int[]{};
 5         //TreeMap要求其key必须可比较大小
 6         TreeMap<Integer, Integer> map = new TreeMap<>((o1,o2) -> o1 - o2);
 7         int[] result = new int[a.length - k + 1];
 8         
 9         for(int i = 0; i < a.length; i++){
10             // 1. add to bst
11             if(map.containsKey(a[i])){
12                 map.put(a[i], map.get(a[i]) + 1 );
13             }else{
14                 map.put(a[i], 1);
15             }         
16             // 2. remove from bst when window sliding
17             if( i >= k){
18                 map.put(a[i - k] , map.get(a[i - k]) - 1 );
19                 if(map.get(a[i - k]) == 0 ){
20                     map.remove(a[i - k]);
21                 }
22             }            
23             // 3. get max
24             if( i + 1 >= k){
25                 result[ i - (k - 1)] = map.lastKey();
26             }           
27         }
28         return result;       
29     }

 

[leetcode]239. Sliding Window Maximum滑动窗口最大值

标签:and   array   取出   you   for   数据   out   span   one   

原文地址:https://www.cnblogs.com/liuliu5151/p/9189712.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!