一个小组的最大实力值

2024-09-03

题目:

给你一个下标从 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;
        }
    }
};