---
title: 一维二维前缀和、差分
date: 2020-03-26 20:44:06
categories:
- 数据结构与算法
tags:
- c++
- 代码模板
- 连续子数组问题
- 二分
---
概述
前缀操作和差分操作是两个互逆操作。求数据区间和能够转化为前缀和的单点操作。对数据区间增减可以转化为
对差分的单点操作,从而提高效率。
但是这种数据结构不支持对一个数据单点操作,并高效地支持区间和操作。要想同时拥有对数据的单点操作和区间和操作,
需要使用树状数组的数据结构。
1. 一维前缀和
前缀和的数据结构能够快速求某个区间的和。方法是在读入数据的时候创建一个数组将0-i的全部数据的和
存储下来。在访问l-r的所有元素的和的时候,只需要返回a[r] - a[l -1]即可。为了方便边界处理,
一般令a[0] = 0。从1开始存数据。
#include
using namespace std;
typedef long long LL;
const int N = 100010;
LL sum[N];
int n, m;
int l, r;
int main(){
cin >> n >> m;
LL temp = 0;
LL in_temp = 0;
for (int i = 1; i < n + 1; i++) {
cin >> in_temp;
temp += in_temp;
sum[i] = temp;
}
for (int i = 1; i < m + 1; i++) {
cin >> l >> r;
cout << sum[r] - sum[l - 1] << endl;
}
return 0;
}
2. 二维前缀和
二维前缀和能够和一维前缀和类似,定义为0, 0 到m, n的矩形内的所有的元素和。
在计算的时候可以直接利用下面的公式来计算
a[i][j] = a[i - 1][j] + a[i][j-1] - a[i-1][j-1]
在访问x1, y1 到x2, y2中所有数据的和的时候只需要返回
a[x2][y2] - a[x2][y1 - 1] - a[x1-1][y1] + a[x1-1][y1-1]即可。
#include
using namespace std;
typedef long long LL;
const int N = 1010;
int num[N][N];
LL sum[N][N];
int main(){
int n, m, q;
int x1, y1, x2, y2;
int t;
cin >> n >> m >> q;
for(int i = 1; i <= n; i ++)
for(int j = 1; j <= m; j ++){
cin >> t;
num[i][j] = t;
}
for(int i = 1; i <= n; i++)
for(int j = 1; j <= m; j++){
sum[i][j] = num[i][j] + sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1];
}
for(int i = 0; i < q; i ++){
cin >> x1 >> y1 >> x2 >> y2;
cout << sum[x2][y2] - sum[x1-1][y2] - sum[x2][y1-1] + sum[x1-1][y1-1] << endl;
}
return 0;
}
3. 一维差分
差分操作可以看做是对前缀和操作的逆操作。可以看做对输入数据,看其是哪个数据的前缀和。
对数据进行差分处理可以将对原始数据的区间加减转化为对差分数据的单点操作。这样可以降低复杂度。
二维同样也可以定义差分,满足对一片元素的操作。
#include
using namespace std;
typedef long long LL;
const int N = 100010;
int n, m;
int q[N];
LL tmp[N];
int l, r, c;
int main(){
cin >> n >> m;
for (int i = 1; i <= n; i ++){
cin >> q[i];
tmp[i] = q[i] - q[i-1];
}
for (int i = 0; i < m; i ++ ){
cin >> l >> r >> c;
tmp[l] += c;
tmp[r + 1] -= c;
}
LL out_temp = 0;
for (int i = 1; i <= n; i ++){
out_temp += tmp[i];
cout << out_temp << " ";
}
return 0;
}
4. 二维差分
#include
typedef long long LL;
using namespace std;
const int N = 1010;
int n,m,q;
LL d[N][N];
LL b[N][N];
void add(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(){
int tmp;
int x1, y1, x2,