[亚马逊面试OA2] Window Sum

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

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

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

[亚马逊面试OA2] Window Sum_第1张图片
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比数组的长度大,该返回什么。

你可能感兴趣的:([亚马逊面试OA2] Window Sum)