problem
题意:先求出数组的总和,然后维护一个当前数组之和curSum,然后对于遍历到的位置,用总和减去当前数字,看得到的结果是否是curSum的两倍,是的话,那么当前位置就是中枢点,返回即可;否则就将当前数字加到curSum中继续遍历,遍历结束后还没返回,说明没有中枢点,返回-1即可。
solution:
注意,不管有几种方式,最后的结果都是从最左边开始的那个结果,那么我们就直接从最左边开始计算,返回第一个满足条件的即可。
class Solution { public: int pivotIndex(vector<int>& nums) { int sum = accumulate(nums.begin(), nums.end(), 0);//err.. int curSum = 0; for(int i=0; i<nums.size(); ++i) { if(sum-nums[i] == 2*curSum) return i; curSum += nums[i]; } return -1; } };
参考
1. Leetcode_easy_724. Find Pivot Index;
2. Grandyang;
完