LeetCode 算法：寻找数组的中心下标

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第2天，点击查看活动详情

寻找数组的中心下标

原题地址

给你一个整数数组 nums ，请计算数组的 中心下标 。

数组 中心下标 是数组的一个下标，其左侧所有元素相加的和等于右侧所有元素相加的和。

如果中心下标位于数组最左端，那么左侧数之和视为 0 ，因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。

如果数组有多个中心下标，应该返回 最靠近左边 的那一个。如果数组不存在中心下标，返回 -1 。

示例 1：

输入：nums = [1, 7, 3, 6, 5, 6]
输出：3
解释：
中心下标是 3 。
左侧数之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 ，
右侧数之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ，二者相等。

示例 2：

输入：nums = [1, 2, 3]
输出：-1
解释：
数组中不存在满足此条件的中心下标。

示例 3：

输入：nums = [2, 1, -1]
输出：0
解释：
中心下标是 0 。
左侧数之和 sum = 0 ，（下标 0 左侧不存在元素），
右侧数之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。

提示：

• 1 <= nums.length <=$10^4$
• -1000 <= nums[i] <= 1000

思路分析

1. 计数组所有项的总和为 total，那么，在遍历到数组 第 i 项 的时候，左侧的和为 leftSum，那么右侧的和为 total - leftSum - nums[i]；
2. 题目要求左侧和右侧的和相同，那么按照上述的分析，在遍历到数组 第 i 项 的时候，会存在 leftSum = total - leftSum -nums[i];
3. 将上述公式进行计算可得知：2 * leftSum + nums[i] = total；
4. 那么，遍历数组，按照这个公式判断，若存在则返回 i， 若遍历完成后没有对应的 i 返回，则返回 -1.

AC 代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var pivotIndex = function(nums) {
    let leftSum = 0
    const total = nums.reduce((prev, curr) => prev + curr, 0)
    for(let i = 0; i < nums.length; i++) {
        if(2 * leftSum + nums[i] === total) {
            return i
        }
        leftSum += nums[i]
    }
    return -1
};

结果：

• 执行结果： 通过
• 执行用时：56 ms, 在所有 JavaScript 提交中击败了98.82%的用户
• 内存消耗：43.3 MB, 在所有 JavaScript 提交中击败了59.61%的用户
• 通过测试用例：745 / 745

END