LeetCode—— 区域和检索 - 数组不可变

题目:

给定一个整数数组  nums,求出数组从索引 到 j  (i ≤ j) 范围内元素的总和,包含 i,  j 两点。

示例:

给定 nums = [-2, 0, 3, -5, 2, -1],求和函数为 sumRange()

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

说明:

  1. 你可以假设数组不可变。
  2. 会多次调用 sumRange 方法。

思路:

最简单的思路就是每次调用sumRange都根据i和j来对数组切片,求和,但是这样在多次调用sumRange时,会重复计算一些数据,浪费时间。所以最快速的方法,应该求所有[0,i]区间的和,然后[i,j] = [0,j] - [0, i- 1],这样只用做一次减法运算。

代码

class NumArray {

    private int[] sums;

    public NumArray(int[] nums) {
        int len = nums.length;
        sums = new int[nums.length+1];
        if (nums.length == 0) {
            return;
        }
        int sum = 0;
        this.sums[0] = 0;
        for(int i = 0;i<len;i++){
            sum += nums[i];
            this.sums[i+1] = sum;
        }
    }
    
    public int sumRange(int i, int j) {
        return sums[j+1] - sums[i];
    }
}
发布了111 篇原创文章 · 获赞 20 · 访问量 13万+

猜你喜欢

转载自blog.csdn.net/qq_23418043/article/details/86507262