题目:
分析:
- 这道题是求某个区间内数组的和, 所以可以用前缀和算法
- 假设i位置是中心位置, 我们要知道[0,i - 1]的和 和 [i + 1,n - 1]的和 看是否相等
- 先求[0,i - 1]的和 用一个前缀数组f
此时f[i] 表示: 前i - 1个数的和, 那么f[i - 1] 就表示前i - 2的和, 所以让f[0] = 0, i从1开始, 从前往后遍历
f[i] = f[i - 1] + nums[i - 1] - 求[i + 1,n - 1]的和 用一个后缀数组g
此时g[i] 表示: 后i + 1个数的和, 那么g[i + 1] 就表示后i + 2的和, 所以让g[n - 1] = 0, i从n-2开始, 从后往前遍历
g[i] = g[i + 1] + nums[i + 1] - 从头遍历数组, 判断f[i] 是否等于 g[i]
代码:
java
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;
}
}