1. 程式人生 > 其它 >[LeetCode] #303 區域和檢索 - 陣列不可變

[LeetCode] #303 區域和檢索 - 陣列不可變

[LeetCode] #303 區域和檢索 - 陣列不可變

給定一個整數陣列 nums,求出陣列從索引 i 到 j(i ≤ j)範圍內元素的總和,包含 i、j 兩點。

實現 NumArray 類:

NumArray(int[] nums) 使用陣列 nums 初始化物件

int sumRange(int i, int j) 返回陣列 nums 從索引 i 到 j(i ≤ j)範圍內元素的總和,包含 i、j 兩點(也就是 sum(nums[i], nums[i + 1], ... , nums[j]))

輸入:
["NumArray", "sumRange", "sumRange", "sumRange"]
[[[-2, 0, 3, -5, 2, -1]], [0, 2], [2, 5], [0, 5]]
輸出:


[null, 1, -1, -3]
解釋:
NumArray numArray = new NumArray([-2, 0, 3, -5, 2, -1]);
numArray.sumRange(0, 2); // return 1 ((-2) + 0 + 3)
numArray.sumRange(2, 5); // return -1 (3 + (-5) + 2 + (-1))
numArray.sumRange(0, 5); // return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))

最直接的方法就是存nums,每次sumRange都計算範圍內和為多少,但不利於多次呼叫,會超出時間限制

所有選擇在初始化時就計算 從nums[0]到當前節點的和,呼叫sumRange只需要減取左邊不需要的和

class NumArray {

    private int[] sums;

    public NumArray(int[] nums) {
        sums = new int[nums.length+1];
        for (int i = 0; i < nums.length; i++) sums[i + 1] = sums[i] + nums[i];
    }
    
    public int sumRange(int left, int right) {
        return sums[right + 1] - sums[left];
    }
}

知識點:

總結: