定义:
arr[]为一个数组 sum[i]为 arr 0到i的前缀和
sum[i]=sum[i-1]+arr[i],i>0
sum[0]=arr[0],i=0
要求L到R的区间和
sum[L,R]等于sum[R]-sum[L-1] (L>0)
sum[L,R]=sum[R] (L=0)
代码如下:
#include<iostream>
#include<Windows.h>
using namespace std;
const int n=5;
int sum[n];
int get_sum(int L,int R){
if(L==0)
return sum[R];
else
return sum[R]-sum[L-1];
}
int main(){
int arr[n]={1,3,6,8,5};
sum[0]=arr[0];
for(int i=0;i<n;i++){
sum[i]=sum[i-1]+arr[i];
}
cout<<get_sum(2,4)<<endl;
cout<<get_sum(0,3)<<endl;
cout<<get_sum(3,4)<<endl;
system("pause");
return 0;
}
运行结果:
首先要明白使用一维数组前缀和的目的是为了降低时间复杂度。例如题目要求sum[3]和sum[4]。
sum[3]=1+3+6+8
sum[4]=1+3+6+8+5
可以看到在求sum[4]时,又重新把1 3 6 8加了一遍,时间复杂度增加。