Leetcode-3427变长子数组求和

依旧前缀和，3427. 变长子数组求和

是一个套路，数组有len个元素，设置一个len+1个元素的数组cnt，cnt[0] = 0,cnt[i+1] = cnt[i]+nums[i]，避开了left为0时需要讨论的情况。left = x, right = y,最后结果 = cnt[y+1] - left[x],可以画个图理解下更直观。

C++代码

class Solution {
public:
    int subarraySum(vector<int>& nums) {
        int len =nums.size();
        vector<int> cnt;
        cnt.resize(len+1);
        int ans = 0;
        for(int i = 0; i <len ;i++){
            cnt[i+1] = cnt[i]+nums[i];
        }
        for(int i = 0; i<len;i++){
            ans+=cnt[i+1]-cnt[max(0,i-nums[i])];
        }
        return ans;
    }
};