前缀和 后缀和 --- 寻找数组的中心下标
题目链接
寻找数组的中心下标
给你一个整数数组 nums ,请计算数组的 中心下标 。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1 。
问题分析
从中⼼下标的定义可知,除中⼼下标的元素外,该元素左边的「前缀和」等于该元素右边的「后缀和」。
▪ 因此,我们可以先预处理出来两个数组,⼀个表⽰前缀和,另⼀个表⽰后缀和。
▪ 然后,我们可以⽤⼀个 for 循环枚举可能的中⼼下标,判断每⼀个位置的「前缀和」以及「后缀和」,如果⼆者相等,就返回当前下标。
代码解决
public int pivotIndex(int[] nums){int n = nums.length;int[] f = new int[n];int[] g = new int[n];//预处理前缀和 后缀和数组for (int i = 1; i <n ; i++)f[i] = f[i-1] + nums[i-1];for (int i = n-2;i >= 0; i--)g[i] = g[i+1] + nums[i+1];//使用for (int i = 0; i < n; i++) {if (f[i] == g[i])return i;}return -1;}