1464. Maximum Product of Two Elements in an Array 数组中两元素的最大乘积

题目地址:https://leetcode-cn.com/problems/maximum-product-of-two-elements-in-an-array/

题目描述

给你一个整数数组 nums,请你选择数组的两个不同下标 ij,使 (nums[i]-1)*(nums[j]-1) 取得最大值。

请你计算并返回该式的最大值。

示例1:

输入:nums = [3,4,5,2]
输出:12 
解释:如果选择下标 i=1 和 j=2(下标从 0 开始),则可以获得最大值,(nums[1]-1)*(nums[2]-1) = (4-1)*(5-1) = 3*4 = 12 。 

示例2:

输入:nums = [1,5,4,5]
输出:16
解释:选择下标 i=1 和 j=3(下标从 0 开始),则可以获得最大值 (5-1)*(5-1) = 16 。

示例3:

输入:nums = [3,7]
输出:12

提示:

1、 2<=nums.length<=500
2、 1<=nums[i]<=10^3

题目大意

略。

解题方法

暴力

首先想到暴力解法,即两重循环,时间复杂度是 O(N^2),题目给出的数组大小只有 500,因此满足要求。

Python 代码如下:

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        res = 0
        N = len(nums)
        for i in range(N):
            for j in range(i + 1, N):
                res = max(res, (nums[i] - 1) * (nums[j] - 1))
        return res

1 2 3 4 5 6 7 8

耗时:932 ms

找最大次大

题目给出的数字全部是正数,因此要求的乘积最大就是 (最大数 - 1) * (次大数 - 1)。题目化为求最大值和次大值的问题。

第一种想法是排序,时间复杂度是 O(N * log(N)).

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        nums.sort()
        return (nums[-2] - 1) * (nums[-1] - 1)

1 2 3 4

耗时:52 ms

第二种想法是遍历一遍,时间复杂度是 O(N)

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        first = 0
        second = 0
        for num in nums:
            if num > first:
                second = first
                first = num
            elif first >= num > second:
                second = num
        return (first - 1) * (second - 1)

1 2 3 4 5 6 7 8 9 10 11

耗时:40 ms

DDKK.COM 弟弟快看-教程,程序员编程资料站,版权归原作者所有

本文经作者:负雪明烛 授权发布,任何组织或个人未经作者授权不得转发