本节博客是用前缀和算法图解“寻找数组的中心下标”,有需要借鉴即可。
1.题目
题目链接:LINK
2.题意
我们以示例1为例来图解一下题意:
3.前缀和求解
根据已有经验,我们想到了用前缀和算法来进行求解:
注:如何想到用前缀和算法的,与之前的题目想法类似,链接:LINK
具体如下:
根据上面图解,我们可以发现f(i) &&g(i)的计算公式为:
f[i] = f[i-1] + arr[i-1]
g[i] = g[i+1] + arr[i+1]
4.示例代码
class Solution { public: int pivotIndex(vector<int>& nums) { int n = nums.size(); vector<int>f(n), g(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; } };
5.细节
思考:为什么我们仅仅最大计算到n-1的和,而没有计算一整个数组的总和?
答案:我们压根不会用到整个数组的和,因为如果要用到整个数组的和,那么中心数组必须得是n,可是原数组没有n这个下标元素!!!
6.总结
这个题目较为简单,不过要想清楚预处理前缀和数组的过程逻辑。
EOF