题目:
给你一个下标从 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;
}
}
网友评论