给出一个m * n的矩阵a,矩阵元素a[i,j]小于1000,进行q次查询,每次查询给出子矩阵的4个边界(上下左右),求该子矩阵所有元素之和。
样例中第一个查询:1 3 1 2 表示从第1行到第3行,从第1列到第2列,对应的子矩阵是:
1 2
5 6
9 10
求和等于33
Input
第一行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)
Output
输出共q行,对应q个询问的求和结果。
经典的二维前缀和问题 , 如果用暴力求解 ,每次查询都会进行一次遍历,时间慢。但如果用二维前缀和的方法进行实现,在查询之前进行前缀和预处理,在查询时的时间复杂度时O(1),效率更高
#include<iostream>
using namespace std;
int main()
{
int arr[100][100] = { 0 }, s[100][100] = {0}, n, m, t;
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
cin >> arr[i][j];
s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + arr[i][j];
}
}
cin >> t;
while (t--)
{
int row1, row2, col1, col2;
cin >> row1 >> row2 >> col1 >> col2;
int num = s[row2][col2] - s[row2][col1-1] - s[row1-1][col2] + s[row1-1][col1-1];
cout << num << endl;
}
return 0;
}