1567. 乘积为正数的最长子数组长度
给你一个整数数组 nums
,请你求出乘积为正数的最长子数组的长度。
一个数组的子数组是由原数组中零个或者更多个连续数字组成的数组。
请你返回乘积为正数的最长子数组长度。
示例 1:
输入:nums = [1,-2,-3,4]
输出:4
解释:数组本身乘积就是正数,值为 24 。
示例 2:
输入:nums = [0,1,-2,-3,-4]
输出:3
解释:最长乘积为正数的子数组为 [1,-2,-3] ,乘积为 6 。
注意,我们不能把 0 也包括到子数组中,因为这样乘积为 0 ,不是正数。
示例 3:
输入:nums = [-1,-2,-3,0,1]
输出:2
解释:乘积为正数的最长子数组是 [-1,-2] 或者 [-2,-3] 。
示例 4:
输入:nums = [-1,2]
输出:1
示例 5:
输入:nums = [1,2,3,5,-6,4,0,10]
输出:4
提示:
-
1 <= nums.length <= 10^5
-
-10^9 <= nums[i] <= 10^9
先来看看这个题的官方题解把:传送门
其中positive[i] 表示以下标 i 结尾的乘积为正数的最长子数组长度,negative[i] 表示乘积为负数的最长子数组长度。
class Solution {
public:
int getMaxLen(vector<int>& nums) {
int n = nums.size();
vector<int> f(n), g(n);
if (nums[0] > 0) f[0] = 1;
else if (nums[0] < 0) g[0] = 1;
int res = f[0];
for (int i = 1; i < n; i ++ ) {
if (nums[i] > 0) {
f[i] = f[i - 1] + 1;
if (g[i - 1]) g[i] = g[i - 1] + 1;
} else if (nums[i] < 0) {
if (g[i - 1]) f[i] = g[i - 1] + 1;
g[i] = f[i - 1] + 1;
}
res = max(res, f[i]);
}
return res;
}
};