九章算法 | Amazon 面试题:滑动窗口内数的和

本文介绍了一道Amazon的面试题,题目要求在给定的整型数组和滑动窗口大小k的情况下,计算滑动窗口内的数之和。通过前缀和和滑动窗口的方法,可以在O(n)的时间复杂度和O(n)的空间复杂度内解决问题。在线评测平台LintCode提供了题目测试。详细解析包括如何利用前缀和的概念以及滑动窗口技巧来求解此问题。
摘要由CSDN通过智能技术生成

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

在线评测地址:LintCode 领扣

样例 1

输入:array = [1,2,7,8,5], k = 3
输出:[10,17,20]
解析:
1 + 2 + 7 = 10
2 + 7 + 8 = 17
7 + 8 + 5 = 20 

算法:前缀和,滑动窗口

  • 前缀和:

前缀和就是一个数列的前n项和,前缀和的计算非常简单,我们只需要额外开一个数组来记录它就好了。每次有新的元素出现我们令​sum[i] = sum[i - 1] + nums[i]​,其中sum是我们用来记录前缀和的数组,nums是我们要输入的数列,i代表数列nums中的第i个数。这样的话我们仅仅需要O(N)的复杂度就可以完成计算,在查询宽度为k的区间之间的值的时候,只需要计算​sum[i] - sum[i - k] + nums[i-k]​即可

  • 滑动窗口:

可以理解成是一个会滑动的窗口,每次记录下窗口的状态,再找出符合条件的适合的窗口,这题要求长度为k,我们维护一个区间[l,r],先计算这个区间的和,然后每次l,r都向右移动一个单位,并减去nums[l],加上nums[r+1],得到新的区间的和。

复杂度分析

  • n为数组的大小
  • 时间复杂度​O(n)​
  • 空间复杂度​O(n)​
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 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值