题目
给出一个 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。
思路
二维前缀和
代码
#include<iostream>
#include<stdio.h>
#include<vector>
using namespace std;
int main()
{
int n,m, num;
cin >> n >> m;
vector<vector<int>> matrix(n+1, vector<int>(m+1, 0));
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= m; j++)
{
cin >> num;
matrix[i][j] = num;
matrix[i][j] += matrix[i-1][j];
matrix[i][j] += matrix[i][j-1];
matrix[i][j] -= matrix[i-1][j-1];
}
}
int q, x1, x2, y1, y2, ans;
scanf("%d", &q);
while(q--)
{
scanf("%d %d %d %d", &x1, &y1, &x2, &y2);
ans = matrix[x2][y2] - matrix[x1-1][y2] - matrix[x2][y1-1] + matrix[x1-1][y1-1];
printf("%d\n", ans);
}
return 0;
}