前缀和和差分是两种预处理方法。
都需要新建一个数组
#include<iostream>
using namespace std;
int main()
{
int a[100] = { 0 };
int b[100] = { 0 };
int c[100] = { 0 };
//前缀和
for (int i = 1; i <= 100; i++)
{
cin >> a[i];
b[i] = a[i] + b[i - 1];
}
//差分
for (int j = 1; j <= 100; j++)
{
cin >> a[j];
c[j] = a[j] - a[j - 1];
}
}
前缀和相对容易理解,差分的意义是什么?
差分的操作核心是求出源数组每两个数据之间的差,类似于记录增降得数据。
当我们需要将一个数组得一段数据都加上一个数,比如a——b之间每一个数据都需要加上一个x,我们只需要对差分数组a加上x,对差分数组b+1得地方减去一个x,就可以完成处理。
前缀和差分数组还有二维的:
如下
#include<iostream>
using namespace std;
const int N = 1000;
int a[N][N], s[N][N];
int main()
{
int n, m, q;
cin >> n >> m >> q;
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
scanf("%d", &a[i][j]);
s[i][j] = s[i][j - 1] + s[i - 1][j] - 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;
}
这个就是对一个数组进行了前缀和处理,并输出了前缀和数组。
二维差分:
#include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int a[N][N], s[N][N];
int main()
{
cin >> n >> m >> q;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
scanf("%d", &s[i][j]);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
a[i][j] = s[i][j] - s[i - 1][j] - s[i][j - 1] + s[i - 1][j - 1];
while (q--)
{
int x1, y1, x2, y2, c;
scanf("%d%d%d%d%d", &x1, &y1, &x2, &y2, &c);
a[x1][y1] += c;
a[x1][y2 + 1] -= c;
a[x2 + 1][y1] -= c;
a[x2 + 1][y2 + 1] += c;
}
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
s[i][j] = a[i][j] + s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++) printf("%d ", s[i][j]);
cout << endl;
}
return 0;
}
return code;