题目:
给定一个整数数组 nums ,找出一个序列中乘积最大的连续子序列(该序列至少包含一个数)
思路:
这题最开始想的是用动态规划来解,但由于正负数的情况不同,定义了positive和negative两个DP数组。
非常非常简单粗暴的思路和各种判断:
import sys
class Solution:
def maxProduct(self, nums: List[int]) -> int:
n = len(nums)
positive = [False for _ in range(n)]
negative = [False for _ in range(n)]
positive_max = [0 for _ in range(n)]
negative_min = [0 for _ in range(n)]
ans = -sys.maxsize
if nums[0] == 0:
positive[0] = True
negative[0] = True
ans = max(ans, nums[0])
elif nums[0] > 0:
positive[0] = True
positive_max[0] = nums[0]
ans = max(ans, nums[0])
elif nums[0] < 0:
negative[0] = True
negative_min[0] = nums[0]
ans = max(ans, nums[0])
for i in range(1, n):
if nums[i] == 0:
positive[i] = True
negative[i] = True
ans = max(ans, nums[i])
elif nums[i] > 0:
positive[i] = True
if positive[i-1]:
positive_max[i] = max(nums[i], nums[i] * positive_max[i-1])
else:
positive_max[i] = nums[i]
ans = max(ans, positive_max[i])
if negative[i-1]:
negative[i] = True
negative_min[i] = nums[i] * negative_min[i-1]
ans = max(ans, negative_min[i])
elif nums[i] < 0:
negative[i] = True
if positive[i-1]:
negative_min[i] = min(nums[i], nums[i] * positive_max[i-1])
else:
negative_min[i] = nums[i]
ans = max(ans, negative_min[i])
if negative[i-1]:
positive[i] = True
positive_max[i] = nums[i] * negative_min[i-1]
ans = max(ans, positive_max[i])
return ans
然而看了题解以后,发现其实不需要用数组记录,由于每个节点只需要它上一个节点的状态,所以可以直接用一个不断更新的变量做记录(也就是positive和negative数组可以简化成一个整数型变量)
其次,我分了positive和negative两种情况,而实际上只需要一个max和min就行,是正是负并没有太大关系。
LeetCode网站题解如下:
class Solution {
public int maxProduct(int[] nums) {
int max = Integer.MIN_VALUE, imax = 1, imin = 1;
for(int i=0; i<nums.length; i++){
//如果是负数,就交换imax和imin
if(nums[i] < 0){
int tmp = imax;
imax = imin;
imin = tmp;
}
imax = Math.max(imax*nums[i], nums[i]);
imin = Math.min(imin*nums[i], nums[i]);
max = Math.max(max, imax);
}
return max;
}
}
作者:guanpengchn
链接:https://leetcode-cn.com/problems/maximum-product-subarray/solution/hua-jie-suan-fa-152-cheng-ji-zui-da-zi-xu-lie-by-g/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。