LeetCode 724. 寻找数组的中心索引

思路

方法：前缀和

记数组的全部元素之和为sum，当遍历到第 i 个元素时，设其左侧元素之和为 tmpSum，则其右侧元素之和为 sum-nums[i]-tmpSum。左右侧元素相等即为 tmpSum = sum-nums[i]-tmpSum。

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int sum = 0;
        for(int i = 0; i < nums.size(); ++i) {
            sum += nums[i];
        }

        int res = -1;
        int tmpSum = 0;
        for(int i = 0; i < nums.size(); ++i) {
            if(tmpSum == sum-nums[i]-tmpSum) {
                res = i;
                break;
            } else {
                tmpSum += nums[i];
            }
        }

        return res;
    }
};

复杂度分析

时间复杂度：O(n)，n为数组长度

空间复杂度：O(1)