239. Sliding Window Maximum
Problem Description
In this problem, you're given an array nums
of integers, and an integer k
which represents the size of a sliding window. This sliding window moves across the array from left to right, one element at a time. At each position of the window, you can only see k
numbers within it. The objective is to return a new array that contains the maximum value from each of the windows as it slides through the entire array.
To illustrate, imagine a section of k
numbers from the array nums
as a frame that only allows you to view those k
numbers. As the frame moves to the right by one element, the oldest number in the frame is replaced with a new number from the array. You need to find the maximum number in the frame after each move and record it.
Intuition
The intuition behind the solution arises from the need to efficiently manage the current candidates for the maximum value within the sliding window. We need a way to update the candidates as the window slides through the array, adding new elements and removing old ones.
One efficient approach to solve this problem is to use a double-ended queue (deque) that maintains a descending order of values (indices of nums
) from the largest at the front to the smallest at the back. The front of the deque will always hold the index of the maximum number within the current window. This allows us to quickly retrieve the maximum number by looking at the element at the front of the deque.
Here are the steps to arrive at the solution:
- Initialize a deque
q
to hold indices ofnums
and an empty listans
to hold the maximums. - Iterate through the indices and values of
nums
using a loop. - Check if the deque is non-empty and if the element at the front of the deque is out of the sliding window's range. If so, pop it from the front to remove it from our candidates.
- While there are elements in the deque and the value at index
i
ofnums
is greater than or equal to the value at the indices stored in the deque's back, pop elements from the back of the deque. This is because the new value could potentially be the new maximum, and thus we remove all the values which are less than the current value since they cannot be the maximum for future windows. - Append the current index
i
to the deque. - If
i
is greater than or equal tok-1
, it means we've filled the first window. The current maximum (the value at the front of the deque) is added toans
. - Continue this process until the end of the array is reached.
- Return the
ans
list.
Using a deque allows us to efficiently remove indices/values from both ends and keep our window updated. This solution ensures that we have a near constant time operation for each window, making it an efficient solution.
Learn more about Queue, Sliding Window, Monotonic Queue and Heap (Priority Queue) patterns.
Solution Approach
The key to the efficient solution is using a double-ended queue (deque), a powerful data structure that allows insertion and deletion from both the front and the back in O(1)
time complexity.
Here's a step-by-step explanation of the algorithm:
-
Initialization: We start by initializing an empty deque
q
and an empty listans
which will store the maximum values of the current sliding window.q = deque() ans = []
-
Iterating through
nums
: We use a for loop to iterate over the array. Each iteration represents a potential new frame of the sliding window:for i, v in enumerate(nums):
-
Maintain Window Size: Before processing the new element at each step, we ensure that the deque's front index belongs to the current window:
if q and i - k + 1 > q[0]: q.popleft()
-
Maintain Deque Properties: We pop indices from the back of the deque as long as the current value is larger than the values at the indices at deque's rear. This ensures that the deque is decreasing so the front always has the largest element:
while q and nums[q[-1]] <= v: q.pop()
-
Add New Index: We then add the index of the current element to the deque back.
q.append(i)
-
Record the Maximum: Once we reach at least the
k
th element (i >= k - 1
), we find a maximum for the full-sized window which we record by adding the value at the front of the deque toans
list:if i >= k - 1: ans.append(nums[q[0]])
-
Return Result: After the loop completes,
ans
contains the maximums for each sliding window, which we return.
This algorithm hinges crucially on the deque's properties to keep the indices in a descending order of their values in nums
. Keeping the deque in descending order ensures that the element at the front is the largest and should be included in ans
. When the window moves to the right, indices that are out of the window or not relevant (because there's a larger or equal value that came later) are popped out from the deque, maintaining the required properties.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Consider the array nums = [1,3,-1,-3,5,3,6,7]
and a sliding window size k = 3
. Let's walk through the solution step by step:
-
Initialization: An empty deque
q
and an empty listans
are initialized.q = deque() # deque to hold indices of nums ans = [] # list to hold max values of each window
-
Iterating through
nums
: We start iterating over the array with indices and values. -
First Iteration (i=0, v=1):
- No indices to remove from deque, since it's empty.
- Append index 0 to the deque.
- Window is not yet full (
i < k-1
), so we don't record the maximum.
-
Second Iteration (i=1, v=3):
- The deque still contains only one index
[0]
. - Since
3
(nums[1]) is greater than1
(nums[q[-1]]), we pop index0
and then append index1
to the deque. - Window is not yet full (
i < k-1
), so we don't record the maximum.
- The deque still contains only one index
-
Third Iteration (i=2, v=-1):
- The deque contains
[1]
and since-1
is not greater than3
, we only append index2
. - Now
i >= k-1
, so we record the current maximumnums[q[0]]
, which is3
.
- The deque contains
-
Subsequent Iterations:
- Continue sliding the window, maintaining the deque by removing out-of-range indices and values smaller than the current one.
- After each slide, if the window is full, add the maximum to
ans
.
-
After Sliding Through Entire Array:
- The final
ans
list will be[3, 3, 5, 5, 6, 7]
.
- The final
This walkthrough demonstrates the solution using a deque to efficiently track the maximum of the current sliding window in the array nums
.
Solution Implementation
1from collections import deque
2
3class Solution:
4 def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
5 # Initialize a deque for storing indices of elements
6 index_queue = deque()
7 # List to store the maximums for each sliding window
8 max_values = []
9
10 # Iterate over each element, with its index
11 for current_index, value in enumerate(nums):
12 # If the first element in the queue is outside the current window, remove it
13 if index_queue and current_index - k + 1 > index_queue[0]:
14 index_queue.popleft()
15
16 # Remove elements from the back of the queue if they are smaller than
17 # or equal to the current element since they will not be needed anymore
18 while index_queue and nums[index_queue[-1]] <= value:
19 index_queue.pop()
20
21 # Add the current index to the queue
22 index_queue.append(current_index)
23
24 # If we have reached or passed the first complete window, record the maximum
25 if current_index >= k - 1:
26 max_values.append(nums[index_queue[0]])
27
28 # Return the list of maximums
29 return max_values
30
1class Solution {
2 public int[] maxSlidingWindow(int[] nums, int k) {
3 int numsLength = nums.length;
4 int[] result = new int[numsLength - k + 1]; // Array to store the max values for each window
5 Deque<Integer> deque = new ArrayDeque<>(); // Double-ended queue to maintain the max element indices
6
7 for (int i = 0, j = 0; i < numsLength; ++i) {
8 // Remove the indices of elements from the deque that are out of the current window
9 if (!deque.isEmpty() && i - k + 1 > deque.peekFirst()) {
10 deque.pollFirst();
11 }
12
13 // Remove indices of elements from the deque that are less than
14 // the current element nums[i] since they are not useful
15 while (!deque.isEmpty() && nums[deque.peekLast()] <= nums[i]) {
16 deque.pollLast();
17 }
18
19 // Add current element's index to the deque
20 deque.offer(i);
21
22 // When we've hit size k, add the current max to the result
23 // This corresponds to the index at the front of the deque
24 if (i >= k - 1) {
25 result[j++] = nums[deque.peekFirst()];
26 }
27 }
28
29 // Return the populated result array containing max of each sliding window
30 return result;
31 }
32}
33
1#include <vector>
2#include <deque>
3using namespace std;
4
5class Solution {
6public:
7 // This function computes the maximum value in each sliding window of size k in the array nums
8 vector<int> maxSlidingWindow(vector<int>& nums, int k) {
9 deque<int> windowIndices; // Deque to store indices of elements in the current window
10 vector<int> maxValues; // Vector to store the maximum value for each window
11
12 for (int i = 0; i < nums.size(); ++i) {
13 // Remove indices of elements not in the current window
14 if (!windowIndices.empty() && i - k >= windowIndices.front()) {
15 windowIndices.pop_front();
16 }
17
18 // Maintain the elements in decreasing order in the deque
19 // Pop elements from the back that are less than or equal to the current element
20 while (!windowIndices.empty() && nums[windowIndices.back()] <= nums[i]) {
21 windowIndices.pop_back();
22 }
23
24 // Push current element's index onto the deque
25 windowIndices.push_back(i);
26
27 // If we've reached the end of the first window, record the max for the current window
28 if (i >= k - 1) {
29 maxValues.emplace_back(nums[windowIndices.front()]);
30 }
31 }
32
33 return maxValues; // Return the list of maximum values
34 }
35};
36
1/**
2 * Calculates the max sliding window for an array of numbers.
3 * @param {number[]} nums - The input array of numbers.
4 * @param {number} k - The size of the sliding window.
5 * @return {number[]} - Array of the maximum numbers for each sliding window.
6 */
7function maxSlidingWindow(nums: number[], k: number): number[] {
8 // Initialize an array to hold the maximum values for each sliding window.
9 let maxValues: number[] = [];
10
11 // Initialize a deque to store indices of elements in nums.
12 // Elements in the deque are in decreasing order from the start (front) to the end (back).
13 let deque: number[] = [];
14
15 // Iterate through nums using 'i' as the right boundary of the sliding window.
16 for (let i = 0; i < nums.length; ++i) {
17 // If the left boundary of the window has exceeded the left-most index in the deque,
18 // remove the left-most index as it's no longer in the window.
19 if (deque.length && i - k + 1 > deque[0]) {
20 deque.shift();
21 }
22
23 // While the deque is not empty and the current element is greater than the
24 // last element indexed in deque, remove the last element from the deque.
25 // This ensures elements in the deque are always in decreasing order.
26 while (deque.length && nums[deque[deque.length - 1]] <= nums[i]) {
27 deque.pop();
28 }
29
30 // Add the current index to the deque.
31 deque.push(i);
32
33 // If we have hit the size of the window, append the maximum value (front of the deque)
34 // to the maxValues array.
35 if (i >= k - 1) {
36 maxValues.push(nums[deque[0]]);
37 }
38 }
39
40 // Return the array containing the maximum for each sliding window.
41 return maxValues;
42}
43
Time and Space Complexity
Time Complexity
The time complexity of the code is O(n)
, where n
is the number of elements in the array nums
. This is because the code iterates through each element in nums
once. Each element is added to the deque q
once, and it's potential removal is also done at most once due to the while loop condition. Despite the nested while loop, the overall number of operations for each element remains constant, ensuring a linear time complexity.
Space Complexity
The space complexity is O(k)
, where k
is the size of the sliding window. The deque q
is used to store indices of elements and its size is restricted by the bounds of the sliding window, which at most can have k
indices stored at a time. Additionally, the output list ans
grows linearly with n - k + 1
(the number of windows), but it does not depend on the input size in a nested or exponential manner, so it does not affect the asymptotic space complexity with respect to k
.
Learn more about how to find time and space complexity quickly using problem constraints.
Which of the two traversal algorithms (BFS and DFS) can be used to find whether two nodes are connected?
Recommended Readings
Queue Intro Think of the last time you stood in line to buy movie tickets The first person in line gets their ticket first and any newcomers join the end of the line This system where the first person to arrive is the first to be served is a queue in real
https algomonster s3 us east 2 amazonaws com cover_photos stack svg Sliding Window Maximum Monotonic Stack We have an array and a sliding window defined by a start index and an end index The sliding window moves from left of the array to right There are always k elements in
Monotonic Stack Deque Intro If you'd prefer a video div class responsive iframe iframe src https www youtube com embed Dq_ObZwTY_Q title YouTube video player frameborder 0 allow accelerometer autoplay clipboard write encrypted media gyroscope picture in picture web share allowfullscreen iframe div The word monotonic means a list or
Want a Structured Path to Master System Design Too? Don’t Miss This!