题解 | #相等的草堆#

#include <numeric>  
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int pivotIndex(vector<int>& nums) {
        // write code here
        if(nums.size()<3)
            return -1;

        int sum = accumulate(nums.begin(), nums.end(), 0);
        int left_sum = 0, right_sum = sum;
        for(int i=1; i+1<nums.size(); ++i)
        {
            left_sum += nums[i-1];
            right_sum = sum-left_sum-nums[i];
            if(left_sum==right_sum)
                return i;
        }

        return -1;
    }
};