算法第三十一天-区域和检索【数组不可变】

区域和检索-数组不可变

题目要求

解题思路

为方便描述，把 $nums$ 记作$a$。
对于数组 $a$，定义它的前缀和
$$\begin{array}{rl}s[0]=& 0\\ s[1]=& a[0]\\ s[2]=& a[0]+a[1]\\ & ⋮\\ s[i]=& a[0]+a[1]+\cdots +a[i-1]=\sum _{j=0}^{i-1}\text{nums}[j]\\ s[i+1]=& a[0]+a[1]+\cdots +a[i-1]+a[i]=\sum _{j=0}^i\text{nums}[j]\\ & ⋮\\ s[n]=& a[0]+a[1]+\cdots +a[n-1]\end{array}$$

根据这个定义，有

$s[i+1]=s[i]+a[i]$
示例中的数组$[-2,0,3,-5,2,-1]$，对应的前缀和数组$s=[0,-2,-2,1,-4,-2,-3]$。
通过前缀和，我们可以把连续子数组的元素和转换成两个前缀和的差，$a[left]$到$a[right]$的元素和等于

$$\sum _{j=\text{left}}^{\text{right}}\text{a}[j]=\sum _{j=0}^{\text{right}}\text{a}[j]-\sum _{j=0}^{\text{left}-1}\text{a}[j]=\text{s}[\text{right}+1]-\text{s}[\text{left}]$$

有了这个式子，示例中子数组 $[3,-5,2,-1]$的和，就可以$O(1)$地用$s[6]-s[2]=-3-(-2)=-1$算出来。

复杂度分析

• 时间复杂度：初始化 $O(n)$，其中 $n$为 $nums$的长度。$sumRangeO(1)$。
• 空间复杂度：$O(n)$。

参考

灵茶山艾府