前缀和
一维前缀和
例题:给一个长度为 n n n的序列 a i ai ai,有 m m m次询问,每次询问 ∑ i = x y a [ i ] \sum_{i=x}^{y}{a[i]} ∑i=xya[i]
分析:如果直接循环求解,那么最多要进行 n ∗ m n*m n∗m次计算。
思考:为什么会算这么多次?——可不可以把区间和记录一下?
解答:定义 p r e [ i ] pre[i] pre[i]表示 ∑ k = 1 i a [ k ] \sum_{k=1}^{i}{a[k]} ∑k=1ia[k]
那么 ∑ i = x y a [ i ] = p r e [ y ] − p r e [ x ] \sum_{i=x}^{y}{a[i]} = pre[y] - pre[x] ∑i=xya[i]=pre[y]−pre[x]
因此我只需要进行 n n n次运算,之后每次都可以一次减法得到答案
pre[1] = a[1];
for(int i = 2; i <= n; i++){//预处理过程
pre[i] = pre[i - 1] + a[i];
}
二维前缀和
有了刚刚的结论,能否扩展到二维?
d p dp dp即之前 p r e pre pre数组 表示从 ( 1 , 1 ) (1,1) (1,1)到 ( i , j ) (i,j) (i,j)的矩形内部的元素和
dp[1][1] = mp[1][1];
for(int i = 1; i <= n; i++){
for(int j = 1; j <= m; j++){
dp[i][j] = mp[i][j] + dp[i - 1][j] + dp[i][j - 1] - dp[i - 1][j - 1];//思考一下矩阵的关系
}
}
int mp[N][N];
int dp[N][N];//从1 1 到 i j 这个矩形内所有数字和 记作 dp[i][j]
int check(int x1, int y1, int x2,int y2){
return dp[x2][y2] + dp[x1 - 1][y1 - 1] - dp[x1 - 1][y2] - dp[x2][y1 - 1];
}