题目链接
思路
- 前缀和的思想
- 第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;
}
}