2018拼多多校招【最大乘积】Python解法

题目描述

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

输入描述:

无序整数数组A[n]

输出描述:

满足条件的最大乘积

思路

由于只需要找出3个数的最大乘积，那么只有两种可能：

1. 最大数、次大数和第三大数的乘积；
2. 最大数、最小数和次小数的乘积(负负得正)。

因此，只需要将输入数字遍历一次，保存最大值、次大值、第三大值以及最小值和次小值即可。

Python源码

n = int(input())
A = list(map(int, input().split()))
max1, max2, max3, min1, min2 = 0, 0, 0, 0, 0
for i in range(n):
    if A[i] > max1:
        max3, max2, max1 = max2, max1, A[i]
    elif A[i] > max2:
        max3, max2 = max2, A[i]
    elif A[i] > max3:
        max3 = A[i]
    elif A[i] < min1:
        min2, min1 = min1, A[i]
    elif A[i] < min2:
        min2 = A[i]
print(max(max1*max2*max3, max1*min1*min2))