左右两边子数组的和相等

左右两边子数组的和相等_第1张图片

        还是累加数组求子数组之和的思想,想求第i个数字左边子数组之和,可以由第i个数字的前缀和减去第i个数字的值,想求第i个数字右边子数组之和,可以由数组所有数字之和减去第i个数字的前缀和。代码如下:

    public static int pivotIndex(int[] nums) {
        int total = 0;
        //先求出整个数组之和
        for (int num : nums) {
            total += num;
        }
        //记录前缀和
        int sum = 0;

        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            //边扫描边判断,扫描到第i个数字,其前缀和便可知
            //其左右两边子数组也就可以计算了
            //找到即可直接返回索引(题目说存在多个返回最左边的数字下标)
            if (sum - nums[i] == total - sum) {
                return i;
            }
        }
        //不存在返回-1
        return -1;
    }

你可能感兴趣的:(算法,算法,数据结构)