from __future__ import annotations from collections import deque arr = [1, 3, -1, -3, 5, 3, 6, 7] window_size = 3 expect = [3, 3, 5, 5, 6, 7] def max_sliding_window(arr: list[float], window_size: int) -> list[float]: """ 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. Each time the sliding window of length window_size moves right by one position. Return the max sliding window. >>> max_sliding_window(arr, window_size) == expect True """ max_val = [] mono_queue: deque = deque() for i in range(len(arr)): # pop the element if the index is outside the window size k if mono_queue and i - mono_queue[0] >= window_size: mono_queue.popleft() # keep the queue monotonically decreasing # so that the max value is always on the top while mono_queue and arr[i] >= arr[mono_queue[-1]]: mono_queue.pop() mono_queue.append(i) # the maximum value is the first element in queue if i >= window_size - 1: max_val.append(arr[mono_queue[0]]) return max_val if __name__ == "__main__": from doctest import testmod testmod()