leetcode 5500. 乘积为正数的最长子数组长度（C++）

给你一个整数数组 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

C++

在到达0之前，记录第一个负数的索引，计算包含当前索引的最大长度

class Solution {
public:
    int getMaxLen(vector<int>& nums) 
    {
        int ans=0;
        int start=0;
        int neg_num=0;
        int neg_idx=-1;
        int n=nums.size();
        for(int i=0;i<n;i++)
        {
            if(0==nums[i])
            {
                start=i+1;
                neg_num=0;
                neg_idx=-1;
                continue;
            }
            else if(nums[i]>0)
            {
                if(neg_num%2==0)
                {
                    ans=max(ans,i-start+1);
                }
                else
                {
                    if(neg_num>0)
                    {
                        ans=max(ans,i-neg_idx);
                    }
                }
            }
            else
            {
                neg_num++;
                if(-1==neg_idx)
                {
                    neg_idx=i;
                }
                if(neg_num%2==0)
                {
                    ans=max(ans,i-start+1);
                }
                else
                {
                    ans=max(ans,i-neg_idx);
                }
            }
        }
        return ans;
    }
};