leetcode--数组中两元素的最大乘积

 题目是LeetCode第191场周赛的第一题,链接:1464. 数组中两元素的最大乘积。具体描述为:给你一个整数数组 nums,请你选择数组的两个不同下标ij,使(nums[i]-1)*(nums[j]-1)取得最大值。请你计算并返回该式的最大值。

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

 示例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

 题目很简单,其实就是让我们找数组中第一大和第二大的两个数罢了,直接一遍遍历就可以完成。时间复杂度为 O ( n ) O(n) O(n),空间复杂度为 O ( 1 ) O(1) O(1)

 JAVA版代码如下:

class Solution {
    public int maxProduct(int[] nums) {
        int first = 0, second = 0;
        for (int n : nums) {
            if (n >= first) {
                int tmp = first;
                first = n;
                second = tmp;
            }
            else if (n >= second) {
                second = n;
            }
        }
        return (first - 1) * (second - 1);
    }
}

 提交结果如下:


 Python版代码如下:

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

 提交结果如下:


你可能感兴趣的:(LeetCode)