给你一个下标从 0 开始的整数数组 nums ,请你找到 最左边 的中间位置 middleIndex (也就是所有可能中间位置下标最小的一个)。

中间位置 middleIndex 是满足 nums[0] + nums[1] + … + nums[middleIndex-1] == nums[middleIndex+1] + nums[middleIndex+2] + … + nums[nums.length-1] 的数组下标。

如果 middleIndex == 0 ,左边部分的和定义为 0 。类似的,如果 middleIndex == nums.length - 1 ,右边部分的和定义为 0 。

请你返回满足上述条件 最左边 的 middleIndex ,如果不存在这样的中间位置,请你返回 -1 。
示例1:

输入:nums = [2,3,-1,8,4]
输出:3
解释:
下标 3 之前的数字和为:2 + 3 + -1 = 4
下标 3 之后的数字和为:4 = 4

示例2:

输入:nums = [1,-1,4]
输出:2
解释:
下标 2 之前的数字和为:1 + -1 = 0
下标 2 之后的数字和为:0

示例3:

示例4:

输入:nums = [1]
输出:0
解释:
下标 0 之前的数字和为:0
下标 0 之后的数字和为:0


输入:nums = [2,5]
输出:-1
解释:
不存在符合要求的 middleIndex 。

方法一: 简单暴力

   var findMiddleIndex = function (nums) {
        for(let i = 0; i < nums.length; i++){
          let left = 0;
        let right = 0;
          // 左侧
          for(let j = 0; j < i; j++){
            left += nums[j];
          }
          for(let j = i + 1; j < nums.length; j++){
            right += nums[j];
          }
         
          if(left === right ){
            return i;
          }
        }
        return -1;
      };
    console.log(findMiddleIndex([2,3,-1,8,4]));

方法2: 前缀和
思路

记数组的全部元素之和为total,当遍历到第 ii 个元素时,设其左侧元素之和为 sum,则其右侧元素之和为 total - sum - nums[i]

左右侧元素相等即为total = sum + nums[i] + sum;

当中心索引左侧或右侧没有元素时,即为零个项相加,这在数学上称作「空和」(\text{empty sum}empty sum)。在程序设计中我们约定「空和是零」

 var findMiddleIndex = function (nums) {
        //  0  是设置的初始值,index将从0开始。不设置的话,index从1开始
        const total = nums.reduce((a, b) => a + b, 0);  // 数组求和
        let sum = 0;
        for (let i = 0; i < nums.length; i++) {
          // 右边和= total - 左边和;sum = total - sum - arr[i]
          if (2 * sum + nums[i] === total) {
            return i;
          }
          sum += nums[i];
          console.log(sum);
        }
      
        return -1;
      };
      console.log(findMiddleIndex([2,3,-1,8,4]));
Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐