子矩阵求和

子矩阵求和

给出一个m * n的矩阵a，矩阵元素ai,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个整数表示矩阵的内容ai,j 。(0 <= ai,j <= 1000) 接下来1行，一个数q，对应查询的数量。(1 <= q <= 1000) 接下来q行，每行4个整数，对应矩阵的上下左右边界u,d,l,r。(1 <= u <= d <= n, 1 <= l <= r <= m)

Output

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

Sample Input

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

二维前缀和代码实现

初始化

for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
cin>>a[i][j];
s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];// 初始化前缀和数组
}

计算子矩阵所有数的和

cout<<s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]
+s[x1-1][y1-1]<<endl;

如果所给数据从一开始就在最后输出前 ++x1,++y1,++x2,++y2自加一就行

题目代码实现

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<map>
#include<set> 
#include<string>
#include<stack>
using namespace std;
int main()
{
	int a[101][101],n,m,q,u,d,l,r;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			cin>>a[i][j];
			a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+a[i][j];
		}
	}
	cin>>q;
	while(q--)
	{
		cin>>u>>d>>l>>r;
		cout<<a[d][r]-a[d][l-1]-a[u-1][r]+a[u-1][l-1]<<endl;
	}
	return 0;
}