Leetcode 628 三个数的最大乘积

一、问题描述

  给你一个整型数组 nums ，在数组中找出由三个数组成的最大乘积，并输出这个乘积。

二、示例及约束

示例 1：
输入： nums = [1,2,3]
输出： 6

示例 2：
输入： nums = [1,2,3,4]
输出： 24

示例 3：
输入： nums = [-1,-2,-3]
输出： -6

提示：
● 3 <= nums.length <= $10^4$
● -1000 <= nums[i] <= 1000

三、代码

方法一：排序

class Solution {
public:
    int maximumProduct(vector<int>& nums) {
        int n = nums.size();
        int poscount = 0, negcount = 0;//记录正负的个数
        
        for (int num:nums){
            if (num < 0){
                negcount++;
            }else if (num > 0){
                poscount++;
            }
        }
        
        sort(nums.begin(),nums.end());//对数组进行排序，默认升序
        
        if (nums.size() == 3){//数组长度为3时，直接输出
            return nums[0]*nums[1]*nums[2];
        }else {
            if((poscount == 0) || (negcount == 0) || (negcount == 1)){
            //当只有一个负数或者全都是正（负）数时，输出最大的三个数
                return nums[n-3]*nums[n-2]*nums[n-1];
            }else if (negcount >= 2){
            //当负数有两个以上时，比较最小的两个负数与最大正数的乘积以及最大的三个正数的乘积
                return max(nums[n-3]*nums[n-2]*nums[n-1],nums[0]*nums[1]*nums[n-1]);
            }
        }
        
        return 0; 
    }
};
**将上述分析过后的代码核心进行精简：**
class Solution {
//本质上是把上面的方法经过分析后精简出核心部分
public:
    int maximumProduct(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int n = nums.size();
        return max(nums[0] * nums[1] * nums[n - 1], nums[n - 3] * nums[n - 2] * nums[n - 1]);
        //只需比较（最小的两个数和最大的数，最大的三个数）这两个乘积谁更大即可
    }
};

方法二：线性扫描

class Solution {
//根据上述分析，不用进行排序，只找出对应的五个数也可求解
public:
    int maximumProduct(vector<int>& nums) {
        int min1 = INT_MAX, min2 = INT_MAX;// 最小的和第二小的
        int max1 = INT_MIN, max2 = INT_MIN, max3 = INT_MIN;// 最大的、第二大的和第三大的
        
        for (int x: nums) {
            if (x < min1) {
                min2 = min1;
                min1 = x;
            } else if (x < min2) {
                min2 = x;
            }

            if (x > max1) {
                max3 = max2;
                max2 = max1;
                max1 = x;
            } else if (x > max2) {
                max3 = max2;
                max2 = x;
            } else if (x > max3) {
                max3 = x;
            }
        }

        return max(min1 * min2 * max1, max1 * max2 * max3);
    }
};

四、总结

时间复杂度：
方法一为O(nlogn)，sort排序的时间。
方法二为O(n)，其中 n 是数组 nums 的长度，遍历了1次。
空间复杂度：
方法一为O(logn)，sort排序的开销。
方法二为O(1)。

方法	时间复杂度	空间复杂度
方法一	O($nlogn$)	O(logn)
方法二	O($n$)	O(1)