编程训练--每周一道编程题(十六)

题目:

给你一个整数数组 nums,请你选择数组的两个不同下标 i 和 j使 (nums[i]-1)*(nums[j]-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 。
输入:nums = [1,5,4,5]
输出:16
解释:选择下标 i=1 和 j=3(下标从 0 开始),则可以获得最大值 (5-1)*(5-1) = 16 。

思路:

  • 取前两个元素作为初始的两个值
  • 从下标 2 开始遍历数组
  • a > b 的时候更新 b 值
  • a <= b 的时候更新 a 值

代码实现:

/**
 * @param {number[]} nums
 * @return {number}
 */
var maxProduct = function (nums) {
    var a = nums[0];
    var b = nums[1];

    for (var i = 2; i < nums.length; i++) {
        if (a > b) {
            nums[i] > b && (b = nums[i]);
        } else {
            nums[i] > a && (a = nums[i]);
        }
    }

    return (a - 1) * (b - 1);

};

 

你可能感兴趣的:(编程)