题目来源:
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 得到。