滑动窗口内数的和
题目:
描述
给你一个大小为n的整型数组和一个大小为k的滑动窗口,将滑动窗口从头移到尾,输出从开始到结束每一个时刻滑动窗口内的数的和。
样例
样例 1
输入:array = [1,2,7,8,5], k = 3
输出:[10,17,20]
解析:
1 + 2 + 7 = 10
2 + 7 + 8 = 17
7 + 8 + 5 = 20
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
int len = nums.length;
if(len == 0 || len < k) {
return new int[]{};
}
int ans[] = new int[len - k + 1], cur = 0;
int l = 0, r = k - 1;
int sum = 0;
for(int i = l; i < r; i++) {
sum += nums[i];
}
while(r < len) {
sum += nums[r];
ans[cur++] = sum;
sum -= nums[l];
l++;
r++;
}
return ans;
}
}