题目:
给你一个下标从 0 开始的整数数组 nums
,它表示一个班级中所有学生在一次考试中的成绩。老师想选出一部分同学组成一个 非空 小组,且这个小组的 实力值 最大,如果这个小组里的学生下标为 i0
, i1
, i2
, … , ik
,那么这个小组的实力值定义为 nums[i0] * nums[i1] * nums[i2] * ... * nums[ik]
。请你返回老师创建的小组能得到的最大实力值为多少。
示例 1:
输入:nums = [3,-1,-5,2,5,-9]
输出:1350
解释:一种构成最大实力值小组的方案是选择下标为 [0,2,3,4,5] 的学生。实力值为 3 * (-5) * 2 * 5 * (-9) = 1350 ,这是可以得到的最大实力值。
示例 2:
输入:nums = [-4,-5,-4]
输出:20
解释:选择下标为 [0, 1] 的学生。得到的实力值为 20 。我们没法得到更大的实力值。
提示:
1 <= nums.length <= 13
-9 <= nums[i] <= 9
思路:
这道题实际上是求所有元素都为整数的数组的子序列的最大积,从最大积的正负性入手。
当数组仅有 1 个元素且为负数时,最大积为负数。
当数组不包含正数,且负数元素小于等于 1 个时,最大积为 0。
其他情况下,最大积为正数。那么如何求这个最大积呢?可以将所有非 0 元素求积,如果乘积为正数,则为最大积。如果乘积为负数,则说明乘积中包含奇数个负数,此时将这个乘积除以最大负数则为最大积。
代码:
class Solution {
public:
long long maxStrength(vector<int>& nums) {
int negativeCount = 0, zeroCount = 0, positiveCount = 0;
long long prod = 1;
int maxNegative = INT_MIN;
for (int num : nums) {
if (num < 0) {
negativeCount++;
prod *= num;
maxNegative = max(maxNegative, num);
} else if (num == 0) {
zeroCount++;
} else {
prod *= num;
positiveCount++;
}
}
if (negativeCount == 1 && zeroCount == 0 && positiveCount == 0) {
return nums[0]; // 只有一个整数,那就只能返回它
}
if (negativeCount <= 1 && positiveCount == 0) {
return 0; // 排除了第一种情况,那就是这种极端情况
}
if (prod < 0) {
return prod / maxNegative;
} else {
return prod;
}
}
};