题目:
给你一个下标从 0 开始的整数数组 nums
,它表示一个班级中所有学生在一次考试中的成绩。老师想选出一部分同学组成一个 非空 小组,且这个小组的 实力值 最大,如果这个小组里的学生下标为 i0
, i1
, i2
, ... , ik
,那么这个小组的实力值定义为 nums[i0] * nums[i1] * nums[i2] * ... * nums[ik]
。请你返回老师创建的小组能得到的最大实力值为多少。
解题思路:
作者:力扣官方题解
链接:https://leetcode.cn/problems/maximum-strength-of-a-group/solutions/2896423/yi-ge-xiao-zu-de-zui-da-shi-li-zhi-by-le-o3nd/
来源:力扣(LeetCode)
这道题实际上是求所有元素都为整数的数组的子序列的最大积,从最大积的正负性入手。
当数组仅有 1 个元素且为负数时,最大积为负数。
当数组不包含正数,且负数元素小于等于 1 个时,最大积为 0。
其他情况下,最大积为正数。那么如何求这个最大积呢?可以将所有非 0 元素求积,如果乘积为正数,则为最大积。如果乘积为负数,则说明乘积中包含奇数个负数,此时将这个乘积除以最大负数则为最大积。
代码:
class Solution {
private:
long long ans;
void solve(int i, long long s, const vector<int> &nums) {
if (i == nums.size()) {
ans = max(ans, s);
return;
}
solve(i + 1, s, nums);
solve(i + 1, s == INT64_MIN ? nums[i] : s * nums[i], nums);
}
public:
long long maxStrength(vector<int>& nums) {
ans = INT64_MIN;
solve(0, INT64_MIN, nums);
return ans;
}
};
优化:
class Solution {
public:
long long maxStrength(vector<int>& nums) {
/*ranges::sort(nums);
int n = nums.size();
if(n == 1) return nums[0];
long long ans = 1;
int negativenums = 0;
int positivenums = 0;
for(int i = 0; i < n; ++i)
{
if(nums[i] < 0) ++negativenums;
else if(nums[i] > 0) ++positivenums;
}
if(negativenums <=1 && positivenums < 1)
{
ans = 0;
}
else if(negativenums % 2 == 0)
{
for(int i = 0; i < n; ++i)
{
if(nums[i] != 0) ans *= nums[i];
}
}
else
{
for(int i = 0; i < negativenums - 1; ++i)
{
ans *= nums[i];
}
for(int i = negativenums; i < n; ++i)
{
if(nums[i] != 0) ans *= nums[i];
}
}
return ans;*/
long long mn = nums[0], mx = mn;
for(int i = 1; i < nums.size(); ++i)
{
long long x = nums[i];
long long tmp = mn;
mn = min({mn, x, mn * x, mx * x});
mx = max({mx, x, tmp * x, mx * x});
}
return mx;
}
};