[亚马逊面试OA2] Window Sum

这道题属于送分题,要是做对,没什么用,要是没做对,那就GG。
题目也很简单,抽象出来就是给你一个数组,再给定一个长度,让你算出数组里面,在这个长度下,分别的连续和。
估计看完了也不知道什么意思,看个例子就知道了。

Given array: [1,2,3,4,5,6,7]
Given window: 3
Output: [6,9,12,15,18]

就想成有一个框,长度是3,沿着数组走,每走一格算一个sum出来。

asdf.png

一般来说,想想就能做出来,直接上代码了,没什么好说的。
非要归个类,应该属于Two Pointers(两指针)的情况

public int[] sum(int[] nums, int window) {
    if (nums == null || nums.length == 0) {
        return new int[0];
    }
    if (nums.length <= window) {
        return nums;
    }
    int sum = 0;
    int index = 0;
    int[] result = new int[nums.length - window + 1];

    for (int i = 0; i < window; i++) {
        sum += nums[i];
    }
    result[index++] = sum;
    for (int i = window; i < nums.length; i++) {
        sum -= nums[i - window];
        sum += nums[i];
        result[index++] = sum;
    }
    return result;
}

需要考虑一点的是,万一window比数组的长度大,该返回什么。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容