还是累加数组求子数组之和的思想,想求第i个数字左边子数组之和,可以由第i个数字的前缀和减去第i个数字的值,想求第i个数字右边子数组之和,可以由数组所有数字之和减去第i个数字的前缀和。代码如下:
public static int pivotIndex(int[] nums) {
int total = 0;
//先求出整个数组之和
for (int num : nums) {
total += num;
}
//记录前缀和
int sum = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
//边扫描边判断,扫描到第i个数字,其前缀和便可知
//其左右两边子数组也就可以计算了
//找到即可直接返回索引(题目说存在多个返回最左边的数字下标)
if (sum - nums[i] == total - sum) {
return i;
}
}
//不存在返回-1
return -1;
}