题解 | #相等的草堆#

知识点：前缀和

思路：因为我们需要重复计算数组中的值，那么很容易想到前缀后，前面有题目细说了前缀和思想，这里不再重复

编程语言：java

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param nums int整型一维数组
     * @return int整型
     */
    public int pivotIndex (int[] nums) {
        // write code here
        //定义前缀和 后缀和数组
        int[] pre = new int[nums.length];
        int[] post = new int[nums.length];
        pre[0] = 0;
        for (int i = 1; i < nums.length; i++) {
            pre[i] = pre[i - 1] + nums[i - 1];
        }
        post[nums.length - 1] = 0;
        for (int i = nums.length - 2; i >= 0; i--) {
            post[i] = post[i + 1] + nums[i + 1];
        }

        for (int i = 0; i < nums.length; i++) {
            if (pre[i] == post[i])
                return i;
        }
        return -1;
    }
}