【算法】前缀和——二维前缀和模板题

本节博客是通过——二位前缀和模板题来介绍前缀和二维算法,有需要借鉴即可。

1.题目

题目链接:LINK
在这里插入图片描述

2.暴力求解

这里我们首先想到的就是一个暴力求解的方式,挨个需要的进行遍历就好了嘛~代码很简单,反正八成这牛客不会让你过。。。这里我们就不再赘述了。

3.二维前缀和算法

想一想,我们暴力求解的时间复杂度直接升到(NMQ)了,之所以能上到这个复杂度无非就是算过的数据没记住,第二次一样的数据还是去重新计算重新计算。

到了这个想法,我们可以想到“以空间换时间”的思路去处理这道题。也就是说我们可以借用“二维前缀和算法”来进行求解。

具体步骤如下

  • 1.预处理一个前缀和数组出来:
    在这里插入图片描述
    所以,这个前缀和数组满足下列规律:
    在这里插入图片描述

  • 2.依据相关规律来利用预处理出来的前缀和数组来进行处理:
    任何一个所求结果,均满足下面规律:
    在这里插入图片描述

这样下来,我们就可以写出下面代码了:

3.代码示例

#include <iostream>
using namespace std;
#include<vector>

int main() 
{
    // 1.读入参数
    int n = 0, m = 0, q = 0;
    cin >> n >> m >> q;
    vector<vector<int>> vv(n+1, vector<int>(m+1));
    for(size_t i = 1; i <= n; i++)
    {
        for(size_t j = 1; j <= m; j++)
        {
            cin >> vv[i][j];
        }
    }
    // 2.预处理二维前缀和数组
    vector<vector<long long>> dp(n+1, vector<long long>(m+1));
    for(size_t i = 1; i <= n; i++)
    {
        for(size_t j = 1; j <= m; j++)
        {
            dp[i][j] = dp[i-1][j] + dp[i][j-1] + vv[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;
    }
}
// 64 位输出请用 printf("%lld")

4.总结

这个题目二位前缀和算法的这个公式推导过程比较重要,建议自己梳理一下。
整体上是一个“以空间换时间”的思路。


EOF

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值