二维数组前缀和的运用

题目分析

输入一个n行m列的整数矩阵，再输入q个询问，每个询问包含四个整数x1, y1, x2, y2，表示一个子矩阵的左上角坐标和右下角坐标。对于每个询问输出子矩阵中所有数的和。

输入格式:

第一行包含三个整数n，m，q。（n,m<=100）

接下来n行，每行包含m个整数，表示整数矩阵。

接下来q行，每行包含四个整数x1, y1, x2, y2，表示一组询问。

输出格式：

共q行，每行输出一个询问的结果。

输入样例：

3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4 

输出样例

17
27
21

本题运用的的知识点主要是二维数组前缀和，有关该知识点的详细讲解请见前缀和，下面介绍这道题的两道解法。

题解

1. 暴力枚举法

#include<iostream>
#define N 110
using namespace std;
int str[N][N];
int main(){
    int n,m,q,x,y,x1,y1,sum=0;
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++)
        cin>>str[i][j];
    }
    while(q--){
        cin>>x>>y>>x1>>y1;
        sum=0;
        for(int i=x;i<=x1;i++){//暴力枚举要求和的范围
            for(int j=y;j<=y1;j++){
                sum=sum+str[i][j];
            }
        }
        cout<<sum<<endl;
    }
    return 0;
}

2. 前缀和法

#include<iostream>
#define N 110
using namespace std;
int str[N][N],arr[N][N];//数组str表示原数据，arr数组表示str的前缀和
int main(){
    int n,m,q,x,y,x1,y1;
    cin>>n>>m>>q;//输入数据
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>str[i][j];
            arr[i][j]=str[i][j]+arr[i-1][j]+arr[i][j-1]-arr[i-1][j-1];//求前缀和数组
        }
    }
    while(q--){
        cin>>x>>y>>x1>>y1;
        cout<<arr[x1][y1]-arr[x1][y-1]-arr[x-1][y1]+arr[x-1][y-1]<<endl;
    }
    return 0;
}