题目:题目链接
题意:从一个数组中找到连续的子数组,其所有元素的积最大用MPS表示。
思路:dp。
先看数据量,2*10^4 所以O(n^2)肯定不行。那就是一维dp了。开始我想着设dp[i]为从0到i的MPS,但状态转移方程一直写不出来。看了看提示,设dp[i]为以i结尾的MPS。那么就很好根据前面推出后一个了。要注意因为是积,所以哪怕前面是负数也有可能再与一个负数相乘为正。所以要保存最大和最小值。
具体如下:
设max_product[i]为以i结尾的最大连续子数组积,min_product[i]为以i结尾的最小连续子数组积。其状态转移方程如下:
m
a
x
_
p
r
o
d
u
c
t
[
i
]
=
m
a
x
(
m
a
x
_
p
r
o
d
u
c
t
[
i
−
1
]
∗
n
u
m
s
[
i
]
,
m
a
x
(
m
i
n
_
p
r
o
d
u
c
t
[
i
−
1
]
∗
n
u
m
s
[
i
]
,
n
u
m
s
[
i
]
)
)
max\_product[i] = max(max\_product[i-1]*nums[i],max(min\_product[i-1]*nums[i],nums[i]))
max_product[i]=max(max_product[i−1]∗nums[i],max(min_product[i−1]∗nums[i],nums[i]))
m
i
n
_
p
r
o
d
u
c
t
[
i
]
=
m
i
n
(
m
a
x
_
p
r
o
d
u
c
t
[
i
−
1
]
∗
n
u
m
s
[
i
]
,
m
i
n
(
m
i
n
_
p
r
o
d
u
c
t
[
i
−
1
]
∗
n
u
m
s
[
i
]
,
n
u
m
s
[
i
]
)
)
min\_product[i] = min(max\_product[i-1]*nums[i],min(min\_product[i-1]*nums[i],nums[i]))
min_product[i]=min(max_product[i−1]∗nums[i],min(min_product[i−1]∗nums[i],nums[i]))
为什么要加一个nums[i]呢,因为nums[i]最大时,可以看作只有nums[i]一个元素的数组,其是MPS(M:min or max)。
那么看代码吧:
class Solution {
public:
int maxProduct(vector<int>& nums) {
vector<int>max_product, min_product;
int ans = nums[0];
max_product.push_back(nums[0]);
min_product.push_back(nums[0]);
for (int i = 1; i < nums.size(); ++i) {
max_product.push_back(
max( max_product[i - 1] * nums[i], max(min_product[i - 1] * nums[i], nums[i]) )
);
min_product.push_back(
min( max_product[i - 1] * nums[i], min(min_product[i - 1] * nums[i], nums[i]) )
);
ans = max(max_product[i], ans);
}
return ans;
}
};
时间复杂度:O(n)
空间复杂度:O(n)
看看能不能优化呢?观察发现,我们使用的max_product,min_product都只跟前一个有关,所以不需要用数组保存,用一个变量保存就可以了。看代码:
class Solution {
public:
int maxProduct(vector<int>& nums) {
int max_product, min_product;
int ans = nums[0];
max_product = min_product = nums[0];
for (int i = 1; i < nums.size(); ++i) {
int tmp = max_product * nums[i];
max_product = (
max( tmp , max(min_product * nums[i], nums[i]) )
);
min_product = (
min( tmp , min(min_product * nums[i], nums[i]) )
);
ans = max(max_product, ans);
}
return ans;
}
};
时间复杂度:O(n)
空间复杂度:O(1)
为什么循环里面要有个tmp呢,因为我们是先算出了max_product,已经把原来的max_product覆盖掉了,而求min_product是需要上一个的max_prodct的。所以需要一个临时变量存储max_product * nums[i]的值。
基本就是这些了。。
加油加油加油加油!!!你在一点一点变强哦,千万别放弃!!!!