前缀和模板
用途
快速查询区间和
时间复杂度 O ( n ) O(n) O(n) 建前缀数组, O ( 1 ) 查询 O(1)查询 O(1)查询
整数二分算法模板
java 代码
一维
//建立s[i]为前缀和数组
for(int i = 1; i <= n; ++i){
s[i] = s[i-1] + a[i];
}
//询问l - r 的区间和
a[r]-a[l-1]
二维
//预处理二维前缀和(s[i][j] -> 1-i , 1- j 的元素总和)
for(int i = 1;i <= n; ++i) {
for(int j = 1; j <= m; ++j) {
s[i][j] = s[i-1][j] + s[i][j-1] -s[i-1][j-1]+a[i][j];
}
}
//询问x1 - x2 , y1 - y2的区间元素总和
s[x2][y2] - s[x1-1][y2] - s[x2][y1-1] + s[x1 - 1][y1-1]
差分算法模板
用途
快速改变一段区间
时间复杂度 O ( 1 ) O(1) O(1) 改变差分数组, O ( n ) O(n) O(n)得到原数组
java 代码
一维
//改变差分数组
void insert(int l, int r, int c)
{
b[l] += c;
b[r + 1] -= c;
}//b[l] += c <=> 原数组后所有元素+c;
//得到原数组
for (int i = 1; i <= n; i ++ ) {
b[i] += b[i-1];
}
二维
void init (int x1,int y1,int x2,int y2,int c) {
a[x1][y1] += c;
a[x2+1][y1] -=c;
a[x1][y2+1] -= c;
a[x2+1][y2+1] += c;
}//差分a[i][j] += c 为将原数组i - n,j - n 总和元素,都加c
//得到原数组
for(int j = 1; j <= m; ++j) {
a[i][j] = a[i-1][j] + a[i][j-1] -a[i-1][j-1] +a[i][j];
}