阵列中两个no的最大乘积(GeeksforGeeks)

def max_product(arr):
    max1 = float('-inf')  # 最大数
    max2 = float('-inf')  # 次大数
    min1 = float('inf')   # 最小数
    min2 = float('inf')   # 次小数

    for num in arr:
        if num > max1:
            max2 = max1
            max1 = num
        elif num > max2:
            max2 = num

        if num < min1:
            min2 = min1
            min1 = num
        elif num < min2:
            min2 = num

    return max(max1 * max2, min1 * min2)

arr = [1, 2, 3, 4, 5]
print(max_product(arr))  # 输出：20