现在有一道题:
输入一个长度为 \(n\) 的整数序列。
接下来再输入 \(m\) 个询问,每个询问输入一对 \(l, r\)。
对于每个询问,输出原序列中从第 \(l\) 个数到第 \(r\) 个数的和。
数据范围
\(1 \le l \le r \le n\)
\(1 \le n,m \le 100000\)
\(-1000 \le 数列中元素的值 \le 1000\)
这种题大家一看就知道打暴力,确实是个好方法 :
那么我们该如何优化它呢?
我们维护一个名为 \(s\) 的前缀和数组,且 \(s[i] = \sum\limits _ {j = 1}^{i} a[j]\),如下:
for (int i = 1; i <= n; ++i)
s[i] = s[i - 1] + a[i];
如果我们维护的序列是 \(1,2,3,4,5,6,7,8,9,10\) 的话,那么我们的前缀和数组为 \(1,3,6,10,15,21,28,36,45,55\) ,那我们该如何利用前缀和数组的性质求和呢?
可以看下面的图:
假如说我们要求 \([l, r]\) 数值的和,我们只需要将 \(s[r]\) (红色部分)减去 \(s[l - 1]\) (蓝色部分)即可。代码:
printf("%d\n", s[r] - s[l - 1]);
完整代码:
#include <cstring>
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N];
int s[N];
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i)
scanf("%d", &a[i]);
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\n", s[r] - s[l - 1]);
}
return 0;
}
题目:
输入一个 \(n\) 行 \(m\) 列的整数矩阵,再输入 \(q\) 个询问,每个询问包含四个整数 \(x_1, y_1, x_2, y_2\),表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。
数据范围
\(1 \le n,m \le 1000\)
\(1 \le q \le 200000\)
\(1 \le x_1 \le x_2 \le n\)
\(1 \le y_1 \le y_2 \le m\)
\(-1000 \le 矩阵内元素的值 \le 1000\)
二维前缀和,可以用来求一个矩阵里人任意一个子矩阵内数的和。
我们用 \(s[I][J]\) 表示 \(\sum\limits_{i = 1}^{I} \sum\limits_{j = 1}^{J} a[i][j]\)。
所以\(s[I][J](绿色部分) = s[I - 1][J](蓝色部分) + s[I][J - 1](红色部分) - s[I - 1][J - 1] (紫色部分)+ a[I][J]\)
查询的话,我们将上面所有颜色的框改一下位置就好了;假如我们要求 \((x_1, y_1)\) 到 \((x_2, y_2)\) 的和,即:\(s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]\)
代码:
#include <cstring>
#include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int a[N][N];
int 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[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]);
}
return 0;
}