要求给定一个数组,获取数组的一个分界点的下标,使得该分界点的两侧子数组的和相等。如果存在多个分界点,只返回第一个分界点的下标,如果没有则返回-1。 要求:时间复杂度不得超过O(n)。

  • Post author:
  • Post category:其他


分析:正常的计算思路应该是,用一个index标记当前计算的下标,从数组第二个元素(即index为1开始)开始计算,将这个元素左边的相加求和,然后右边相加求和,对比是否相等。然后index++依次遍历,知道index 遍历到数组倒数第二个元素,都没有找到则返回-1。

  static int middleIndexWithArray(Integer[] array) {
        if (array.length < 3) return -1;
        int allSum = 0;
        int leftSum = 0;
        int rightSum = 0;
        for (int i = 0; i < array.length; i++) {
            allSum += array[i];
        }
        for (int i = 1; i < array.length - 1; i++) {
            leftSum += array[i - 1].intValue();
            rightSum = allSum - leftSum - array[i].intValue();
            if (leftSum == rightSum) return i;
        }
        return -1;
    }

开始需要比较的是array数组下标为1的元素,这时如果知道了整个数组的和为allSum。那么左边子数组的和leftSum就是array的第一个元素array[0]。右边的子数组的和就是rightSum = allSum – leftSum – array[1];

当下一次index2时,将上一次计算的leftSum用于下一次比较,leftSum只需要再加上array[1],就是左边数组的和,而右边的子数组的和还可以通过rightSum = allSum – leftSum – array[1]计算。

这样的话,左侧数组和只需要每次累加一个新的需要对比的元素,右侧直接就可以通过减法计算不需要累加: