小猴编程周赛C++ | 2022年春第15周T3子矩阵的和

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】

有一个$n\times m$的矩阵，第i行第j列格子的值为$a_{i,j}$，现在小猴有q个问题 – 给出两个正方形区域（第k个正方形左上角格子在第$x_k$行第$y_k$列，边长为$le{n}_k$），请问被区域覆盖的子矩阵中的所有的元素之和是多少？

【输入】

第一行两个正整数n，m；

接下来n行，每行m格整数$a_{i,1},\dots ,a_{i,m}$；

接下来一行一个正整数q；

接下来q行每行六个整数$x_1,y_1,le{n}_1,x_2,y_2,le{n}_2$。

【输出】

共q行，每行一个整数表示结果。

【输入样例】

2 2
1 2 
3 4
3
1 1 1 2 2 1
1 2 1 2 1 1
2 1 1 2 2 1

【输出样例】

5
5
7

【代码详解】

#include <bits/stdc++.h>
using namespace std;
int x[5], y[5];
long long a[510][510];

long long area(int x1, int y1, int x2, int y2)
{
    long long sum = 0;
    for (int i=x1; i<=x2; i++) {
        sum += a[i][y2] - a[i][y1-1];  // 二维前缀和优化
    }
    return sum;
}
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];
            a[i][j] += a[i][j-1];
        }
    }
    int q;
    cin >> q;
    while (q--) {
        int x1, y1, x2, y2, len1, len2;
        cin >> x1 >> y1 >> len1 >> x2 >> y2 >> len2;

        x[1] = x1;
        y[1] = y1;
        x[2] = x[1] + len1 - 1;
        y[2] = y[1] + len1 - 1;
        x[3] = x2;
        y[3] = y2;
        x[4] = x[3] + len2 - 1;
        y[4] = y[3] + len2 - 1;
        long long ans = 0;
        ans += area(x[1], y[1], x[2], y[2]);
        ans += area(x[3], y[3], x[4], y[4]);
        if (!(x[2]<x[3] || y[2]<y[3] || x[1]>x[4] || y[1]>y[4])) {
            sort(x+1, x+4+1);
            sort(y+1, y+4+1);
            ans -= area(x[2], y[2], x[3], y[3]);
        }
        cout << ans << endl;
    }
    return 0;
}

【运行结果】

2 2
1 2
3 4
3
1 1 1 2 2 1
5
1 2 1 2 1 1
5
2 1 1 2 2 1
7