You are given an array of integers 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 1:
Input: nums = [1,3,-1,-3,5,3,6,7], 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
Example 2:
Input: nums = [1], k = 1
Output: [1]
Constraints:
-   1 <= nums.length <= 105
-   -104 <= nums[i] <= 104
-   1 <= k <= nums.length
SOLUTION:
import heapq
class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        n = len(nums)
        maxes = []
        deleted = {}
        window = [-num for num in nums[:k]]
        heapq.heapify(window)
        maxes.append(-window[0])
        for i in range(0, n - k):
            heapq.heappush(window, -nums[i + k])
            deleted[-nums[i]] = deleted.get(-nums[i], 0) + 1
            curr = window[0]
            while curr in deleted:
                heapq.heappop(window)
                deleted[curr] -= 1
                if deleted[curr] == 0:
                    del deleted[curr]
                curr = window[0]
            maxes.append(-curr)
        return maxes
 

 
    
Top comments (0)