题面:
给出一个 nn 行 mm 列的矩阵,矩阵的每个位置有一个非负整数 a[i][j]a[i][j],有 qq 次询问,每次询问求一个左上角为 (a,b)(a,b),右下角为 (c,d)(c,d) 的子矩阵的所有数之和。
输入格式
第一行两个整数 n,mn,m,表示矩阵的行和列的大小。
接下来 nn 行每行 mm 个整数,为矩阵内容。
接下来一行为一个整数 qq ,表示询问次数。
接下来 qq 行每行 44 个整数 a,b,c,da,b,c,d,含义见题面。
输出格式
共 qq 行,第 ii 行为第 ii 个询问的答案。
数据范围
n \times m \le 100,000n×m≤100,000,a[i][j] \le 1000a[i][j]≤1000,q \le 100,000q≤100,000,1 \le a \le c \le n1≤a≤c≤n,1 \le b \le d \le m1≤b≤d≤m。
Sample 1
Input
3 5
1 2 3 4 5
3 2 1 4 7
2 4 2 1 2
3
1 1 3 5
2 2 3 3
1 1 3 3
Output
43
9
20
思路:这道题很容易想到容斥原理,一般的思路是建立一个二维数组,a[i][j],表示(1,1)与(i,j)所围成区域的矩阵和。题面询问(x1, y1)(x2, y2)所围成区域矩阵和,就是a[x2][y2] - a[x2][y1 - 1] - a[x1 - 1][y2] + a[x1 - 1][y1 - 1]的结果
但是数据范围不允许我们直接开二维数组,我们可以想到二维数组在内存其实是连续存储的,所以我们可以用一维数组来模拟二维数组,用乘法运算来处理换行
另外要注意的就是我们模拟的二维数组是存在第0行和第0列的,所以在一维数组的序列上就是应该将行列加1,具体看代码实现
AC代码
#include <iostream>
#include <cstdio>
using namespace std;
const int N = 1e6;
int s[N + 5];
int n, m, t, a, b, c, d, x, y;
int f(int x, int y)
{
return x * m + y + 1;
}
int main()
{
scanf("%d%d", &n, &m);
++n, ++m;
for(int i = 1; i < n; ++i)
for(int j = 1; j < m; ++j)
{
scanf("%d", &s[f(i, j)]);
s[f(i, j)] += s[f(i - 1, j)] + s[f(i, j - 1)] - s[f(i - 1, j - 1)];
}
scanf("%d", &t);
for(int i = 1; i <= t; ++i)
{
scanf("%d%d%d%d", &a, &b, &c, &d);
int r1 = s[f(c, d)], r2 = s[f(c, b - 1)] + s[f(a - 1, d)] - s[f(a - 1, b - 1)];
printf("%d\n", r1 - r2);
}
return 0;
}