leetcode:1464. 数组中两元素的最大乘积(python3解法）

心软且酷丶2024-01-03 20:03

难度：简单

给你一个整数数组 nums，请你选择数组的两个不同下标 i 和 j，使 (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

提示：

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

`题解：`

python 复制代码

class Solution(object):
    import itertools
    def maxProduct(self, nums):
        list1 = []
        list2 = []
        res = []
        for i in range(len(nums)):
            list1.append(i)
            list2.append(i)
        combinations = list(itertools.product(list1, list2))
        # print(combinations)
        for i in combinations:
            if i[0] == i[1]:
                continue
            res.append((nums[i[0]]-1)*(nums[i[1]]-1))
        return max(res)