leetcode 724. 寻找数组的中心索引

题目链接
https://leetcode-cn.com/problems/find-pivot-index/

一开始自己想到的思路是前缀和后缀和然后求相等值的位置，看了题解发现是自己笨比了。nums[i]前面的元素和*2+nums[i]=元素总和 的i即为所求。另外个人觉得一个坑点在于默认没有元素的一边总和为0。

代码如下。

import java.util.Arrays;

public class Problem724 {
    public int pivotIndex(int[] nums) {
        int total= Arrays.stream(nums).sum();
        int sum=0;
        for (int i = 0; i < nums.length; i++) {
            if (sum*2+nums[i]== total)
                return i;
            sum+=nums[i];
        }
        return -1;
    }
}