剑指 Offer II 012. 左右两边子数组的和相等

题目链接

思路

• 前缀和的思想
• 第0个元素左边的数组和为0，最后一个元素右边的元素默认为0

代码

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