【一次过】Lintcode 604. 滑动窗口内数的和

最新推荐文章于 2021-07-30 10:25:11 发布

原创最新推荐文章于 2021-07-30 10:25:11 发布 · 248 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 专栏收录该内容

393 篇文章

订阅专栏

本文介绍了一种解决滑动窗口求和问题的算法。给定一个整型数组和一个固定大小的滑动窗口，算法将窗口从数组头部移动到尾部，计算并输出每个位置上窗口内元素的总和。示例展示了对于数组[1,2,7,8,5]和窗口大小3，算法将返回[10,17,20]作为结果。

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

给你一个大小为n的整型数组和一个大小为k的滑动窗口，将滑动窗口从头移到尾，输出从开始到结束每一个时刻滑动窗口内的数的和。

样例

对于数组 [1,2,7,8,5] ，滑动窗口大小k= 3 。
1 + 2 + 7 = 10
2 + 7 + 8 = 17
7 + 8 + 5 = 20
返回 [10,17,20]

解题思路：

直接遍历即可。注意返回空数组是返回new int[0]

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