152. 乘积最大子序列
给定一个整数数组 nums ,找出一个序列中乘积最大的连续子序列(该序列至少包含一个数)。
解题思路: 如果此题使用暴力解法解题的话,最简单的方法是做两层循环选取子序列,然后计算子序列的乘积,综合起来,时间复杂度为
O
(
N
3
)
O(N^3)
O(N3),肯定会TLE,然后想到用DP解题,
d
p
(
i
,
j
)
dp(i,j)
dp(i,j)表示从i到j序列的最大乘积,动态转移方程为
d
p
(
i
,
j
)
=
m
a
x
(
d
p
(
i
+
1
,
j
)
∗
A
i
,
d
p
(
i
,
j
−
1
)
∗
A
j
)
dp(i,j)=max(dp(i+1,j)*A_i,dp(i,j-1)*A_j)
dp(i,j)=max(dp(i+1,j)∗Ai,dp(i,j−1)∗Aj),但是时间复杂度为
O
(
N
2
)
O(N^2)
O(N2),提交OJ后依然TLE,然后参考了网友Grandyang,其实此题依然是用DP解题。动态转移方程看那篇博客不难知道,定义
m
x
(
i
)
mx(i)
mx(i)表示从i前面某处到i处这段序列的的最大乘积,
m
n
(
i
)
mn(i)
mn(i)表示从i前面某处到i处这段序列的的最小乘积,最大值和最小值会从
m
x
(
i
−
1
)
∗
A
i
,
m
n
(
i
−
1
)
∗
A
i
,
A
i
mx(i-1)*A_i,mn(i-1)*A_i,A_i
mx(i−1)∗Ai,mn(i−1)∗Ai,Ai这三个值中选。现在我们探讨一下为什么是这样,因为题目要求的是选取连续序列,因此以
A
i
A_i
Ai为终点的序列的最大值,一定是有他本身和他前一个极值决定的,由于负数的存在,极小值乘上负数有可能变为极大值,因此这个最大值可能从这三个数中产生。请看代码。
class Solution {
public:
int maxProduct(vector<int>& nums) {
int n = nums.size();
vector<int> mx(n, INT_MIN), mn(n, INT_MAX);
mx[0] = nums[0];
mn[0] = nums[0];
int res = nums[0];
for (int i = 1; i < n; ++i) {
int val1 = mx[i - 1] * nums[i], val2 = mn[i - 1] * nums[i], val3 = nums[i];
mx[i] = max(val1, max(val2, val3));
mn[i] = min(val1, min(val2, val3));
res = max(res, mx[i]);
}
return res;
}
};