题解 | #相等的草堆#

import java.util.*;


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

本题主要考察的知识点就是数组前缀和，所用编程语言是java。

我们利用前缀和数组，即一个数组pre，其中pre[i]表示grass[0]到grass[i]的重量之和。例如，如果grass = [2, 4, 6, 8, 10]，那么pre = [2, 6, 12, 20, 30]。然后，我们可以遍历grass数组，对于每个草堆i，我们可以计算左边草堆的总重量为pre[i-1]（如果i > 0），右边草堆的总重量为pre[n-1] - pre[i]（如果i < n-1）。如果这两个值相等，那么我们就找到了一个符合条件的草堆，并返回它的下标i。