LintCode - Window Sum

最新推荐文章于 2024-04-14 09:56:25 发布

原创最新推荐文章于 2024-04-14 09:56:25 发布 · 298 阅读

0 ·

CC 4.0 BY-SA版权

LintCode 专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种高效的滑动窗口求和算法实现，通过一次遍历即可计算出数组中每个长度为k的子数组的元素之和。该算法首先计算初始窗口的元素总和，然后通过每次移除窗口左侧元素并加入右侧新元素的方式，更新窗口内元素的总和，最终返回所有窗口的元素和组成的数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    /**
     * @param nums: a list of integers.
     * @param k: length of window.
     * @return: the sum of the element inside the window at each moving.
     */
    vector<int> winSum(vector<int> &nums, int k) {
        // write your code here
        int n = nums.size();
        if(n<k || k==0) return {};
        int sum = 0;
        for(int i=0;i<k;i++){
            sum += nums[i];
        }
        vector<int> res{sum};
        int l=1, r=k;
        while(r<n){
            sum = sum-nums[l-1]+nums[r];
            res.push_back(sum);
            l++;
            r++;
        }
        return res;
    }
};