[亞馬遜面試OA2] Window Sum

這道題屬于送分題,要是做對(duì),沒什么用,要是沒做對(duì),那就GG。
題目也很簡單,抽象出來就是給你一個(gè)數(shù)組,再給定一個(gè)長度,讓你算出數(shù)組里面,在這個(gè)長度下,分別的連續(xù)和。
估計(jì)看完了也不知道什么意思,看個(gè)例子就知道了。

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

就想成有一個(gè)框,長度是3,沿著數(shù)組走,每走一格算一個(gè)sum出來。

asdf.png

一般來說,想想就能做出來,直接上代碼了,沒什么好說的。
非要?dú)w個(gè)類,應(yīng)該屬于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;
}

需要考慮一點(diǎn)的是,萬一window比數(shù)組的長度大,該返回什么。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容