0
点赞
收藏
分享

微信扫一扫

B - 前后缀——子矩阵求和

ZGtheGreat 2022-01-07 阅读 42

B - 前后缀——子矩阵求和

给出一个m * n的矩阵a,矩阵元素a[i,j]小于1000,进行q次查询,每次查询给出子矩阵的4个边界(上下左右),求该子矩阵所有元素之和。

样例中第一个查询:1 3 1 2 表示从第1行到第3行,从第1列到第2列,对应的子矩阵是:

1 2
5 6
9 10

求和等于33

第一行2个整数n, m,中间用空格分割,分别对应数组的行数n、列数m(1 <= m,n <= 100) 接下来n行,每行m个整数表示矩阵的内容a[i,j] 。(0 <= a[i,j] <= 1000) 接下来1行,一个数q,对应查询的数量。(1 <= q <= 1000) 接下来q行,每行4个整数,对应矩阵的上下左右边界u,d,l,r。(1 <= u <= d <= n, 1 <= l <= r <= m)

输出共q行,对应q个询问的求和结果。

3 4
1 2 3 4
5 6 7 8
9 10 11 12
3
1 3 1 2
1 2 1 3
1 3 1 3

Sample Output

33
24
54

Sponsor

//跟第一题一模一样
#include<iostream>
#include<algorithm>

using namespace std;
int a[1005][1005]={0};

int main() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
cin >> a[i][j];
}
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
a[i][j] += a[i][j - 1] + a[i - 1][j] - a[i - 1][j - 1];
}
}
int q;
cin >> q;
int u, d, l, r;
int answer[1005] = { 0 };
int ans = 0;
for (int k = 0; k < q; k++) {
cin >> u >> d >> l >> r;
u--; l--;
answer[ans++] = a[d][r] - a[d][l] - a[u][r] + a[u][l];
}
for (int h = 0; h < ans; h++) {
cout << answer[h];
if (h < ans - 1) {
cout << endl;
}
}
return 0;
}
举报

相关推荐

0 条评论