题目来源:
leetcode题目,网址:LCR 012. 寻找数组的中心下标 - 力扣(LeetCode)
解题思路:
获取第一个元素左边的元素 leftSum 和 右边的元素和 rightSum,按序遍历数组的同时不断更新 leftSum 和 rightSum 直至二者相等或完成遍历。
解题代码:
class Solution {
public int pivotIndex(int[] nums) {
int leftSum=0;
int rightSum=Arrays.stream(nums).sum()-nums[0];
for(int i=0;i<nums.length;i++){
if(leftSum==rightSum){
return i;
}
leftSum+=nums[i];
if(i<nums.length-1)
rightSum-=nums[i+1];
}
return -1;
}
}
总结:
官方题解采用前缀和的思想。此外 rightSum 可通过 sum-nums[i]-leftSum 得到。