寻找数组的中心下标 ---- 前缀和

题目接

题目:

分析:

• 这道题是求某个区间内数组的和, 所以可以用前缀和算法
• 假设i位置是中心位置, 我们要知道0,i - 1的和 和 i + 1,n - 1的和 看是否相等
• 先求0,i - 1的和 用一个前缀数组f
  此时fi 表示: 前i - 1个数的和, 那么fi - 1 就表示前i - 2的和, 所以让f0 = 0, i从1开始, 从前往后遍历
  fi = fi - 1 + numsi - 1
• 求i + 1,n - 1的和 用一个后缀数组g
  此时gi 表示: 后i + 1个数的和, 那么gi + 1 就表示后i + 2的和, 所以让gn - 1 = 0, i从n-2开始, 从后往前遍历
  gi = gi + 1 + numsi + 1
• 从头遍历数组, 判断fi 是否等于 gi

代码:

class Solution {
    public int pivotIndex(int[] nums) {
        int n = nums.length;
        int[] f = new int[n];
        int[] g = new int[n];
        for (int j = 1; j < n; j++) {
            f[j] = f[j - 1] + nums[j -1];
        }
        for (int j = n - 2; j >= 0; j--) {
            g[j] = g[j + 1] + nums[j + 1];
        }
        for (int i = 0; i < n; i++) {
            if(f[i] == g[i]){
                return i;
            }
        }
        return -1;
    }
}