[Leetcode724] 力扣724:寻找数组的中心索引

在这里插入图片描述
思路
有关子数组和的问题可以思考能否用前缀和方法解决

class Solution {
    
    
public:
    int pivotIndex(vector<int>& nums) {
    
    
        int sum = 0;
        for (int& n : nums) {
    
    
            sum += n;
        }
        int preSum = 0;
        for (int i = 0; i < nums.size(); ++i) {
    
    
            if (2 * preSum + nums[i] == sum) return i;
            else preSum += nums[i];
            }
        return -1;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_44537258/article/details/113320581