寻找数组的中心下标

leetcode 724题

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

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

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

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

题目要求下标位置的左侧所有元素之和等于右侧所有元素之和，我们首先定义数组的总和为total，那么对于第i个元素，左边的元素和为sum，右边元素和为total-nums[i]-sum，如果要这两个相等，那么就存在sum=total-nums[i]-sum --> 2*sum+nums[i]=total，从这个思路就可以设计出代码

class Solution {
    public int pivotIndex(int[] nums) {
        int tital=Arrays.stream(nums).sum();
        int sum=0;
        for(int i=0;i<nums.length;i++){
            if(sum*2+nums[i]==total){
                return i;
               }
            sum+=nums[i];
        }
        return -1;
    }
}

那么我们再次思考，如果sum+nums[i]=total-sum，而sum=nums[0]+nums[1]+...+nums[i]，那么在计算的时候我们能不能通过减去计算过的nums的元素来计算右边的和，每当i向右移的时候，就把total减掉nums[i]。

class Solution {
    public int pivotIndex(int[] nums) {
        int total = Arrays.stream(nums).sum();
        int sum = 0;
        for (int i = 0; i < nums.length; i++) {
            sum+=nums[i];
            if (sum == total) {
                return i;
            }
            total-=nums[i];
        }
        return -1;
    }
}