【Leetcode_easy】724. Find Pivot Index

problem

​​724. Find Pivot Index​​

题意：先求出数组的总和，然后维护一个当前数组之和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​​;

完