[LeetCode]1567. 乘积为正数的最长子数组长度

题目

思路

遇0保存当前记录长度后对其清零；

遇负数可以靠后续的负数使其乘积为正，记录长度+1；

           若后续无负数，根据长度判断是抛弃当前负数还是抛弃第一个负数（最后一个负数的后续长度和第一个负数的前置长度）

遇正数不改变乘积结果，记录长度+1；

---------------------------------------------------------------------------------------------------------------------

别人思路：

先用0将nums切分成多个片段，保证在每个片段中只有正数和负数（便于处理）

若片段中负数的个数为偶数个，则最长子数组即为最长片段；

若片段中负数的个数为奇数个，根据长度判断是抛弃最后一个负数还是抛弃第一个负数（最后一个负数的后续长度和第一个负数的前置长度）

---------------------------------------------------------------------------------------------------------------------

具体处理上：因为是遇0切分，所以我在nums最后插入了0；

代码

class Solution:
    def getMaxLen(self, nums: List[int]) -> int:
        nums_chunks = [] # nums被0分成的若干片段
        nums_chunk = [] # nums被0分成的某个片段
        for num in nums:
            if num == 0:
                if nums_chunk:
                    nums_chunks.append(nums_chunk)
                nums_chunk = []
            else:
                nums_chunk.append(num)
        if nums_chunk:
            nums_chunks.append(nums_chunk)
        
        ans = 0
        for nums_chunk in nums_chunks:
            neg_cnt = 0
            first_neg_i = -1 # 第一次出现负数的位置
            last_neg_i = -1 # 最后一次出现负数的位置
            for i, num in enumerate(nums_chunk):
                if neg_cnt == 0: 
                    if num < 0:
                        first_neg_i = i
                        last_neg_i = first_neg_i
                        neg_cnt += 1
                else:
                    if num < 0:
                        last_neg_i = i
                        neg_cnt += 1
            if neg_cnt % 2 == 0: # 如果负数的个数为偶数次，那么乘积是正数，直接返回数组长度
                ans = max(ans, len(nums_chunk))
            else:
                ans = max(ans, len(nums_chunk) - min(first_neg_i + 1, len(nums_chunk) - last_neg_i))
        return ans 

作者：akasumi
链接：https://leetcode.cn/problems/maximum-length-of-subarray-with-positive-product/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

官方思路

动态规划

递推式：“当前乘积为正=前一乘积为正+该数为正 or 前一乘积为负+该数为负”

边界条件：nums长度为1时根据其值判断

----------------------------------------------------------------------------------------------------------------

positive[i]表示以下标 i结尾的乘积为正数的最长子数组长度，negative[i]表示乘积为负数的最长子数组长度。

class Solution:
    def getMaxLen(self, nums: List[int]) -> int:
        length = len(nums)
        positive, negative = [0] * length, [0] * length
        if nums[0] > 0:
            positive[0] = 1
        elif nums[0] < 0:
            negative[0] = 1
        
        maxLength = positive[0]
        for i in range(1, length):
            if nums[i] > 0:
                positive[i] = positive[i - 1] + 1
                negative[i] = (negative[i - 1] + 1 if negative[i - 1] > 0 else 0)
            elif nums[i] < 0:
                positive[i] = (negative[i - 1] + 1 if negative[i - 1] > 0 else 0)
                negative[i] = positive[i - 1] + 1
            else:
                positive[i] = negative[i] = 0
            maxLength = max(maxLength, positive[i])

        return maxLength

作者：力扣官方题解
链接：https://leetcode.cn/problems/maximum-length-of-subarray-with-positive-product/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

总结

起先我是写了一版代码，并通过不断调试通过了94/112个案例，但在某个案例上反复调试不通后我去看了评论区的解法，再看我自己的代码就发现这是什么垃圾...遂删除记录。

其实之前做过动态规划的题，但光分析题目的时候还是想不起来...