最大乘积(拼多多2018校招)

题目描述

给定一个无序数组，包含正数、负数和0，要求从中找出3个数的乘积，使得乘积最大，要求时间复杂度：O(n)，空间复杂度：O(1)

输入描述

无序整数数组A[n]

输出描述

满足条件的最大乘积

示例1

输入：3 4 1 2

输出：24

算法分析

记录数组中第一大max1，第二大max2，第三大max3以及最小min1，次小min2的五个数，那么最大乘积为max(max1*max2*max3, max1*min1*min2)，变量要定义为long，否则会溢出。

提交代码：

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

int main()
{
    int m, number;
    long long result;
    vector<int> arr;
    cin >> m;
    for (int j = 0; j < m; ++j) {
        cin >> number;
        arr.push_back(number);
    }
    
    long max1 = 0, max2 = 0, max3 = 0, min1 = 0, min2 = 0;
    for (int i = 0; i < arr.size(); ++i) {
        if (arr[i] == 0) continue;
        if (arr[i] > max1) {
            max3 = max2;
            max2 = max1;
            max1 = arr[i];
        } else if (arr[i] > max2) {
            max3 = max2;
            max2 = arr[i];
        } else if (arr[i] > max3)
            max3 = arr[i];
        else if (arr[i] < min1)
        {
            min2 = min1;
            min1 = arr[i];
        } else if (arr[i] >= min1 && arr[i] < min2)
            min2 = arr[i];
    }
    
    result = max(max1*max2*max3, max1*min1*min2);
    cout << result << endl;
    return 0;
}