【Lintcode】206. Interval Sum

题目地址：

https://www.lintcode.com/problem/interval-sum/description

给定一个数组，要求设计一个数据结构，实现区间和的查询。

最方便的实现区间和查询的数据结构是前缀和数组。需要注意的是，前缀和数组$S$的长度应该开成原数组的长度多一个，$S[i]$表示原数组$A$的前$i$个数之和，那么$S[0]=0$就表示原数组的前$0$个数之和，也就是$0$。$$\sum _{i=j}^{k}A[i]=S[k+1]-S[j]$$区间和可以在$O(1)$的时间内得到。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /**
     * @param A: An integer list
     * @param queries: An query list
     * @return: The result list
     */
    public List<Long> intervalSum(int[] A, List<Interval> queries) {
        // write your code here
        long[] preSum = new long[A.length + 1];
        // 计算前缀和数组
        for (int i = 0; i < A.length; i++) {
            preSum[i + 1] = preSum[i] + A[i];
        }
        
        List<Long> res = new ArrayList<>();
        for (Interval query : queries) {
            long end = preSum[query.end + 1], start = preSum[query.start];
            res.add(end - start);
        }
        
        return res;
    }
}

class Interval {
    int start, end;
    Interval(int start, int end) {
        this.start = start;
        this.end = end;
    }
}

预处理时间复杂度$O(n)$，空间$O(n)$，查询时空复杂度$O(1)$。