【前缀和】专题 | 【模板】二维前缀和

一、题目链接

【模板】二维前缀和

二、题目

三、算法原理

最差情况，每次询问都要求整个区间的和，遍历一整个区间的时间复杂度是O(nm)，询问q次时间复杂度是O(nmq)。

四、编写代码

#include <iostream>
#include <vector>
using namespace std;
int main()
{
    // 1.读入数据
    int n = 0, m = 0, q = 0;
    cin >> n >> m >> q;
    vector<vector<int>> arr(n + 1, vector<int>(m + 1));
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= m; ++j)
            cin >> arr[i][j];
    
    // 2.预处理一个前缀和矩阵
    vector<vector<long long>> dp(n + 1, vector<long long>(m + 1));// 防止溢出
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= m; ++j)
            dp[i][j] = dp[i - 1][j] + dp[i][j - 1] + arr[i][j] - dp[i - 1][j - 1];
    
    // 3.使用前缀和矩阵
    int x1 = 0, y1 = 0, x2 = 0, y2 = 0;
    while (q--)
    {
        cin >> x1 >> y1 >> x2 >> y2;
        cout << dp[x2][y2] - dp[x1 - 1][y2] - dp[x2][y1 - 1] + dp[x1 - 1][y1 - 1] << endl;
    }
    return 0;
}