【算法题】2708. 一个小组的最大实力值

2023-05-31  本文已影响0人  程序员小2

题目:

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

java代码:

class Solution {
    public long maxStrength(int[] nums) {
        if (nums.length == 1) return nums[0];
        var hasPos = false;
        int negCnt = 0, zeroCnt = 0;
        Arrays.sort(nums);
        long v1 = 1; // 正数的乘积
        long v2 = 1; // 负数的乘积
        for (int x : nums) {
            if (x > 0) { v1 = v1 * x; hasPos = true; }
            else if (x < 0) { negCnt++; }
            else { zeroCnt++; }
        }
        // 返回0的情况
        if (zeroCnt == nums.length || (zeroCnt > 0 && negCnt == 1 && !hasPos)) return 0;
        int idx = 0;
        // 依次选取负数,如果选了奇数个,那么去掉最后一个不选。
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] >= 0) break;
            v2 = v2 * (long) nums[i];
            idx = i;
        }
        if (v2 > 0) return v1 * v2;
        v2 /= nums[idx]; // 去掉最后一个负数不选
        return v1 * v2;
    }
}

上一篇下一篇

猜你喜欢

热点阅读