Skip to main content

寻找数组的中心下标

Tips

题目类型: 前缀和

相关题目:

题目​

给你一个整数数组 nums, 请编写一个能够返回数组"中心下标"的方法. 数组中心下标的定义是:

  • 数组的一个下标, 其左侧所有元素相加的和等于右侧所有元素相加的和
  • 如果数组不存在中心下标, 返回 -1
  • 如果数组有多个中心下标, 应该返回最靠近左边的那一个
示例
输入: nums = [1, 7, 3, 6, 5, 6]
输出: 3
解释: 左侧数之和: 1 + 7 + 3 = 11, 右侧数之和: 5 + 6 = 11
输入: nums = [1, 2, 3]
输出: -1
解释: 数组中不存在满足此条件的中心下标
输入: nums = [2, 1, -1]
输出: 0
解释: 下标 0 左侧不存在元素, 视作和为 0, 右侧数之和为 1 + (-1) = 0

题解​

  1. 先求出整个数组的总和
  2. 做一波遍历, 总和 - 当前元素 - 左边子数组总和 就是右边子数组的总和, 如果左边子数组总和跟右边子数组的总和相等, 也就找到了中心下标, 否则返回 -1.
/**
 * @param {number[]} nums
 * @return {number}
 */
var pivotIndex = function (nums) {
  const len = nums.length
  const sum = nums.reduce((acc, val) => acc + val)
  let leftSum = 0

  for (let i = 0; i < len; i++) {
    if (sum - nums[i] - leftSum === leftSum) {
      return i
    }
    leftSum += nums[i]
  }

  return -1
}