0303. Range Sum Query - Immutable (E)

佚名 4年前 (2020-06-27) 算法 779人围观抢沙发百度已收录

Range Sum Query - Immutable (E)

题目

Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.

Example:

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

Given nums = [-2, 0, 3, -5, 2, -1]

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

Note:

You may assume that the array does not change.
There are many calls to sumRange function.

题意

给定一个数组，求从下标i到j的元素之和。

思路

因为方法可能被多次调用，因而可以用缓存进行优化：另设数组sum，sum[i]表示nums中前i个元素之和。因此i到j的元素之和就可以表示为sum[j + 1] - sum[i]。

代码实现

Java

class NumArray {
    private int[] sum;

    public NumArray(int[] nums) {
        sum = new int[nums.length + 1];
        for (int i = 0; i < nums.length; i++) {
            sum[i + 1] = sum[i] + nums[i];
        }
    }

    public int sumRange(int i, int j) {
        return sum[j + 1] - sum[i];
    }
}