前缀和
前缀和可以理解成一个数组中从第一项加到第n项的和
前缀和的作用:可以计算从第l项到第r项的和。
前缀和能把n次计算简化成一次
#include <iostream>
using namespace std;
const int N = 100010;
int a[N],s[N];
int n,m;
int main(){
scanf("%d%d",&n,&m);
for(int i = 1; i<= n;i++) scanf("%d",&a[i]);
//s[0] = 0;
for(int i = 1;i <= n; i++) s[i] = s[i-1] + a[i]; //前缀和的初始化
while(m--) {
int l,r;
scanf("%d%d",&l,&r);
printf("%d",s[r] - s[l-1]);//区间和的计算
}
return 0;
}
子矩阵的和(二维前缀和)
思路:一图胜千言
就是将区域进行拆分,类似于概率论中的二维随机变量的分布函数那个图哈哈哈。
#include <iostream>
using namespace std;
const int N = 1010;
int m, n, q;
int a[N][N], s[N][N];
int main() {
scanf("%d%d%d", &n, &m, &q);
for (int i = 1; i <= n; i++)//从第一个开始存
for (int j = 1; j <= m; j++)
scanf("%d", &a[i][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]; //求前缀和(根据图形)
while (q--) {
int x1, y1, x2, y2;
scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
//计算部分和
printf("%d\n", s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1]);
}
return 0;
}
时间复杂度小课堂
O(n)就说明 n个数据要运算n次
O(log n)说明 n个数据要运算log以2为底的n次
所以logn快得多,100万个数据,二分法O(logn)只需要20多次计算!!所以logn快得多,100万个数据,二分法O(logn)只需要20多次计算!!
一维差分
假设有两个数组a,b,使得a[i] = b[1]+b[2]+…+b[i]
则称a是b的前缀和数组,b是a的差分数组。
要在a数组de[l,r]区间内每一个元素都加上c,只需要将b[l]+c,b[r+1]-c即可。
默认a,b数组初始元素为零。不用考虑构造的过程,只用考虑更新的过程
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], b[N];//a表示原数组,b表示差分数组
//b数组当做假想的数组,满足a是b的前缀和,不用考虑构造的过程,只用考虑更新的过程
void insert(int l, int r, int c) {
b[l] += c;//l之后的每一个元素都加上了c
b[r + 1] -= c;//r+1之后每一个元素都减去c
//最终只有[l,r]内的元素加上了c,其余元素不变
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
//将a,b数组默认初始全为零,然后在每个区间[i,i]中插入a[i]
for (int i = 1; i <= n; i++) insert(i, i, a[i]);
//执行完这一步操作后,b中的元素与a中元素相同
while (m--) {//进行m次插入操作
int l, r, c;
scanf("%d%d%d", &l, &r, &c);
insert(l, r, c);//在b数组的[l,r]区间插入c
}
for (int i = 1; i <= n; i++) b[i] += b[i - 1]; //将b数组变成自己的前缀和数组,操作完之后b是a的前缀和数组+c
for (int i = 1; i <= n; i++) printf("%d", b[i]);
return 0;
}
二维差分
b[x1][y1] += c;
b[x2+1][y1] -= c;
b[x1][y2+1] -= c;
b[x2+1][y2+1] += c;
# include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int a[N][N], b[N][N];
void insert(int x1, int y1, int x2, int y2, int c) {
b[x1][y1] += c;
b[x1][y2 + 1] -= c;
b[x2 + 1][y1] -= c;
b[x2 + 1][y2 + 1] += c;
}
int main() {
scanf("%d%d%d", &n, &m, &q);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
scanf("%d", &a[i][j]);
//插入a数组
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
insert(i, j, i, j, a[i][j]);
//在[x1,y1]-[x2,y2]之间加上c,进行q次操作
while (q--) {
int x1, y1, x2, y2, c;
cin >> x1 >> y1 >> x2 >> y2 >> c;
insert(x1,y1,x2,y2,c);
}
//求b数组本身的前缀和
for(int i = 1;i<=n;i++)
for(int j = 1;j<= m;j++)
b[i][j] += b[i-1][j] + b[i][j-1] - b[i-1][j-1];
for(int i = 1;i<=n;i++) {
for (int j = 1; j <= m; j++)
printf("%d ", b[i][j]);
printf("\n");
}
}