1567. Maximum Length of Subarray With Positive Product
Problem Description
The given problem presents us with an array of integers, nums
. Our task is to find the length of the longest subarray where the product of all the elements is positive. A subarray is defined as a contiguous part of the original array and can contain zero or more elements. The output should be the length of the longest such subarray.
To solve this problem, we must consider that the product of elements in a subarray is positive if there are an even number of negative numbers (including zero) because the product of two negatives is positive. We must also remember that the presence of a zero resets the product because any number multiplied by zero is zero.
Intuition
The intuition behind the solution is based on dynamic programming. We can keep track of the maximum subarray length with a positive product in two situations:
- The current number is positive
- The current number is negative
We do this through two variables, f1
and f2
, where f1
represents the length of the subarray ending at the current position with a positive product, and f2
(if non-zero) represents the length of a subarray ending at the current position with a negative product.
Here's the approach to arrive at the solution:
- Initialize
f1
andf2
. If the first number is positive,f1
starts at 1, and if it's negative,f2
starts at 1. This sets up our initial condition. - Iterate through the array, starting from the second element, and update
f1
andf2
based on the current number:
- If the current number is positive:
- Increment
f1
because a positive number won't change the sign of the product. - If
f2
is non-zero, increment it as well, as a positive number won't affect the sign of an already negative product.
- Increment
- If the current number is negative:
- Assign
f1
the value off2 + 1
, because a negative number changes the sign of the product. - Assign
f2
the value off1
(before updatingf1
) + 1, again because a negative number changes the sign.
- Assign
- If the current number is zero, reset both
f1
andf2
to zero because the product would be zero regardless of the preceding values.
- At each step, compare
f1
with our current maximum length (res
) and updateres
iff1
is larger. - Once we've iterated through the array, return
res
as it contains the maximum length of a subarray with a positive product.
Thus, using this approach, we are able to avoid calculating the product of subarray elements and can find the maximum length of a subarray with a positive product efficiently.
Learn more about Greedy and Dynamic Programming patterns.
Solution Approach
The solution uses a dynamic programming approach that revolves around two main concepts: tracking the longest subarray length where the product is positive (f1
), and tracking the longest subarray length where the product is negative (f2
). To implement this, we use a simple linear scan of the input array with constant space complexity, updating both f1
and f2
based on the sign and value of the current element.
Here is a step-by-step explanation of the algorithm:
-
Initialize
f1
andf2
:f1
is set to 1 if the first element ofnums
is positive, indicating the subarray starts with a positive product.f2
is set to 1 if the first element ofnums
is negative, indicating the subarray starts with a negative product.res
is initialized tof1
to keep track of the maximum length.
-
Iterate over the array starting from the second element:
- For each element, there are three cases to consider:
a. The current element is positive:
- Increment
f1
as the positive number will not change the sign of the product. - Increment
f2
only if it is positive, meaning there was a previous negative number and this positive number extends the subarray with a now positive product. b. The current element is negative: - Swap
f1
andf2
and incrementf2
, as the negative number makes a previously positive product negative. - Only increment
f1
iff2
was originally positive, as this indicates we had a negative product and now have made it positive. c. The current element is zero: - Reset
f1
andf2
to 0 as any subarray containing a zero has a product of zero, which is neither positive nor negative.
- Increment
- For each element, there are three cases to consider:
a. The current element is positive:
-
After considering the current number, update the maximum length
res
with the current value off1
iff1
is greater than the currentres
. -
Continue the loop until the end of the array.
-
Return the final value of
res
, which contains the maximum length of the subarray with a positive product.
The code uses constant extra space (only needing f1
, f2
, and res
), and it runs in O(n) time, where n is the length of the input array nums
. This is an efficient solution because it avoids the need to calculate products directly and uses the properties of positive and negative numbers to infer the sign of the subarray products.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Let's walk through an example to illustrate the solution approach using the following array of integers:
nums = [1, -2, -3, 4]
Our goal is to find the length of the longest subarray where the product of all elements is positive.
-
Initialize
f1
,f2
, andres
:- The first element is positive, so
f1 = 1
. - There are no negative elements yet, so
f2 = 0
. - Initialize
res
with the value off1
, henceres = 1
.
- The first element is positive, so
-
Iterate over the array:
a. Second element,
-2
(current element is negative):- Since
f1 = 1
andf2 = 0
, we updatef2
tof1 + 1
, resulting inf2 = 2
. - There is no previously recorded negative subarray (
f2
was 0), sof1
remains the same. - No update to
res
asf1
has not increased.
b. Third element,
-3
(current element is negative):- Swap
f1
andf2
. Before the swap,f1 = 1
andf2 = 2
. After the swap,f1
is now 2. - Increment
f2
to becomef1(pre-swap) + 1
, hencef2 = 2
. - Update
res
with the current value off1
which is now2
, as it is greater than the previousres
.
c. Fourth element,
4
(current element is positive):- Increment
f1
tof1 + 1
, sof1
becomes3
. - Increment
f2
as well since it's non-zero, sof2
becomes3
. - Update
res
with the current value off1
which is3
as it is greater than the previousres
.
- Since
-
After iterating through the array, the maximum length
res
is3
. We have found the longest subarray[1, -2, -3, 4]
, which indeed has a positive product and its length is 3.
Therefore, the length of the longest subarray with a positive product is 3
.
Solution Implementation
1from typing import List
2
3class Solution:
4 def getMaxLen(self, nums: List[int]) -> int:
5 # f_positive is the length of the longest subarray with a positive product ending at the current position
6 f_positive = 1 if nums[0] > 0 else 0
7
8 # f_negative is the length of the longest subarray with a negative product ending at the current position
9 f_negative = 1 if nums[0] < 0 else 0
10
11 # res is the length of the longest subarray with a positive product found so far
12 res = f_positive
13
14 # Iterate through the array starting from the second element
15 for num in nums[1:]:
16 # Store previous f_positive and f_negative before updating
17 prev_f_positive, prev_f_negative = f_positive, f_negative
18
19 # When the current number is positive
20 if num > 0:
21 # Extend the subarray with a positive product
22 f_positive += 1
23 # If there was a subarray with a negative product, extend it too; otherwise, reset to 0
24 f_negative = prev_f_negative + 1 if prev_f_negative > 0 else 0
25
26 # When the current number is negative
27 elif num < 0:
28 # The new subarray with a negative product becomes the previous positive subarray plus the current negative number
29 f_negative = prev_f_positive + 1
30 # If there was a subarray with a negative product, it becomes positive now; otherwise, reset to 0
31 f_positive = prev_f_negative + 1 if prev_f_negative > 0 else 0
32
33 # When the current number is zero, reset both counts to 0
34 else:
35 f_positive = 0
36 f_negative = 0
37
38 # Update res to be the max of itself and the current positive subarray length
39 res = max(res, f_positive)
40
41 # After iterating through the array, return the result
42 return res
43
1class Solution {
2
3 public int getMaxLen(int[] nums) {
4 int positiveCount = nums[0] > 0 ? 1 : 0; // Initialize the length of positive product subarray
5 int negativeCount = nums[0] < 0 ? 1 : 0; // Initialize the length of negative product subarray
6 int maxLength = positiveCount; // Store the maximum length of subarray with positive product
7
8 // Iterate over the array starting from the second element
9 for (int i = 1; i < nums.length; ++i) {
10 if (nums[i] > 0) {
11 // If the current number is positive, increase the length of positive product subarray
12 ++positiveCount;
13 // If there was a negative product subarray, increase its length too
14 negativeCount = negativeCount > 0 ? negativeCount + 1 : 0;
15 } else if (nums[i] < 0) {
16 // If the current number is negative, swap the lengths of positive and negative product subarrays
17 int previousPositiveCount = positiveCount;
18 int previousNegativeCount = negativeCount;
19 negativeCount = previousPositiveCount + 1;
20 positiveCount = previousNegativeCount > 0 ? previousNegativeCount + 1 : 0;
21 } else {
22 // If the current number is zero, reset the lengths as any sequence will be discontinued by zero
23 positiveCount = 0;
24 negativeCount = 0;
25 }
26 // Update the maximum length if the current positive product subarray is longer
27 maxLength = Math.max(maxLength, positiveCount);
28 }
29 return maxLength; // Return the maximum length found
30 }
31}
32
1#include <vector> // Include necessary header for vector usage
2#include <algorithm> // Include for the max() function
3
4class Solution {
5public:
6 int getMaxLen(std::vector<int>& nums) {
7 // Initialize lengths of subarrays with positive product (positiveLen) and negative product (negativeLen)
8 int positiveLen = nums[0] > 0 ? 1 : 0;
9 int negativeLen = nums[0] < 0 ? 1 : 0;
10 int result = positiveLen; // This will store the maximum length of subarray with positive product
11
12 // Iterate through the array starting from the second element
13 for (int i = 1; i < nums.size(); ++i) {
14 if (nums[i] > 0) {
15 // If the current number is positive, increment positive length
16 ++positiveLen;
17 // If there was a negative product, increment negative length, otherwise reset to 0
18 negativeLen = negativeLen > 0 ? negativeLen + 1 : 0;
19 } else if (nums[i] < 0) {
20 // Store the previous lengths before updating
21 int prevPositiveLen = positiveLen;
22 int prevNegativeLen = negativeLen;
23
24 // If the current number is negative, the new negative length is
25 // the previous positive length + 1, as it changes the sign
26 negativeLen = prevPositiveLen + 1;
27
28 // The new positive length would be the previous negative length + 1
29 // If there has been no negative number yet, reset positive length to 0
30 positiveLen = prevNegativeLen > 0 ? prevNegativeLen + 1 : 0;
31 } else { // If the current number is 0, reset lengths as the product is broken
32 positiveLen = 0;
33 negativeLen = 0;
34 }
35
36 // Update result if the current positive length is greater
37 result = std::max(result, positiveLen);
38 }
39
40 return result; // Return the maximum length of subarray with positive product
41 }
42};
43
1function getMaxLen(nums: number[]): number {
2 // Initialize counts for positive sequence (posSeqCount) and negative sequence (negSeqCount)
3 let posSeqCount = nums[0] > 0 ? 1 : 0;
4 let negSeqCount = nums[0] < 0 ? 1 : 0;
5
6 // Initialize the answer (maxLen) with the count of the positive sequence
7 let maxLen = posSeqCount;
8
9 // Loop through the array starting from index 1
10 for (let i = 1; i < nums.length; ++i) {
11 let current = nums[i]; // Current number in the array
12
13 if (current === 0) {
14 // If current number is 0, reset the counts
15 posSeqCount = 0;
16 negSeqCount = 0;
17 } else if (current > 0) {
18 // If current number is positive, increment the positive sequence count
19 posSeqCount++;
20 // If there are negative counts, increment it, otherwise set to 0
21 negSeqCount = negSeqCount > 0 ? negSeqCount + 1 : 0;
22 } else {
23 // If current number is negative, swap the counts after incrementing
24 let tempPosSeq = posSeqCount;
25 let tempNegSeq = negSeqCount;
26 // Increment negative sequence count if there's a positive sequence; otherwise set to 0
27 posSeqCount = tempNegSeq > 0 ? tempNegSeq + 1 : 0;
28 // Always increment the negative sequence count since current number is negative
29 negSeqCount = tempPosSeq + 1;
30 }
31
32 // Update maxLen to the greater of maxLen or the current positive sequence count
33 maxLen = Math.max(maxLen, posSeqCount);
34 }
35
36 // Return the maximum length of subarray of positive product
37 return maxLen;
38}
39
Time and Space Complexity
The provided code maintains two variables f1
and f2
to keep track of the length of the longest subarray with a positive product and the length of the longest subarray with a negative product, respectively.
Time Complexity:
The function iterates through the list nums
once. For each element in nums
, it performs a constant number of computations: updating f1
, f2
, and res
based on the sign of the current element. Therefore, the time complexity is O(n)
, where n
is the number of elements in nums
.
Space Complexity:
The space complexity is O(1)
because the code uses a fixed amount of space: variables f1
, f2
, res
, pf1
, and pf2
. The space used does not grow with the size of the input array.
Learn more about how to find time and space complexity quickly using problem constraints.
Consider the classic dynamic programming of longest increasing subsequence:
Find the length of the longest subsequence of a given sequence such that all elements of the subsequence are sorted in increasing order.
For example, the length of LIS for [50, 3, 10, 7, 40, 80]
is 4
and LIS is
[3, 7, 40, 80]
.
What is the recurrence relation?
Recommended Readings
Greedy Introduction div class responsive iframe iframe src https www youtube com embed WTslqPbj7I title YouTube video player frameborder 0 allow accelerometer autoplay clipboard write encrypted media gyroscope picture in picture web share allowfullscreen iframe div When do we use greedy Greedy algorithms tend to solve optimization problems Typically they will ask you to calculate the max min of some value Commonly you may see this phrased in the problem as max min longest shortest largest smallest etc These keywords can be identified by just scanning
What is Dynamic Programming Prerequisite DFS problems dfs_intro Backtracking problems backtracking Memoization problems memoization_intro Pruning problems backtracking_pruning Dynamic programming is an algorithmic optimization technique that breaks down a complicated problem into smaller overlapping sub problems in a recursive manner and uses solutions to the sub problems to construct a solution
LeetCode Patterns Your Personal Dijkstra's Algorithm to Landing Your Dream Job The goal of AlgoMonster is to help you get a job in the shortest amount of time possible in a data driven way We compiled datasets of tech interview problems and broke them down by patterns This way we
Want a Structured Path to Master System Design Too? Don’t Miss This!