c++ vector求和

打工人潮哥

使用 numeric库的accumulate函数，其中accumulate函数的第三个参数为累加的初值。该参数帮助accumulate函数知道将要累加的元素类型。

#include<numeric>
class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        if(nums.size()==0){
            return 0;
        }
        int first_sum = 0;
        int sum = accumulate(nums.begin(),nums.end(),0);
        for(int i=0;i<nums.size();i++){
            if(first_sum == sum-first_sum-nums){
                return i;
            }
            else{
                first_sum += nums;
            }
        }
        return -1;

    }
};

雪花漫飞

学习下