力扣精选算法100道——【模板】前缀和 (二维)

目录

🎈题目解析

🎈算法原理

🎈实现代码


二维前缀和【模板】


🎈题目解析

上一题我们讲述了一维的前缀和求法。

第一行三个参数,n是行数=3,m是列数=4,q=3代表查询次数

接下来就是n行m列的矩阵

剩下的q行 是查询的区间,我们依次输入x1,y1, x2, y2


🎈算法原理

我们先想想如何计算这个正方形整个的区域和。

我们可以利用这种方法 A+B+C+D=(A+B)+(A+C)+D-A

使用两层循环计算前缀和,其中**dp[i][j] 表示矩阵中从 (1,1)(i,j) 的子矩阵的和**。计算公式为**dp[i][j] = dp[i-1][j](A+B) + dp[i][j-1](A+C) + arr[i][j](D) - dp[i-1][j-1](A)。**

dp[i][j] = dp[i-1][j] + dp[i][j-1] + arr[i][j] - dp[i-1][j-1]


通过循环输入查询的矩形区域的左上角和右下角坐标 (x1, y1)(x2, y2)

输出查询区域的和,通过使用前缀和矩阵 dp,计算出区域和为 dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1],这个公式通过前缀和的性质得出。

区域和为ret= dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1]


🎈实现代码

复制代码
#include <iostream>
#include<vector>
using namespace std;

int main()
{
    int n=0,m=0,q=0;
    cin>>n>>m>>q;
    vector<vector<int>>arr(n+1,vector<int>(m+1))(n*m,1);
    for (int i=1;i<=n;i++) 
    {
        for(int j=1;j<=m;j++)
        {
            cin>>arr[i][j];
        }
    }    

    //预处理前缀和公式
      vector<vector<long long>>dp(n+1,vector<long long>(m+1));
    //从(1,1)到(i,j)各个区间前缀和
      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];
        }
      }

    //使用前缀和矩阵
    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;
}

我要一直开心~

相关推荐
晨非辰8 小时前
【数据结构入坑指南】--《层序分明:堆的实现、排序与TOP-K问题一站式攻克(源码实战)》
c语言·开发语言·数据结构·算法·面试
hansang_IR9 小时前
【题解】P2217 [HAOI2007] 分割矩阵 [记忆化搜索]
c++·数学·算法·记忆化搜索·深搜
Voyager_410 小时前
算法学习记录03——二叉树学习笔记:从两道题看透后序位置的关键作用
笔记·学习·算法
我搞slam15 小时前
快乐数--leetcode
算法·leetcode·哈希算法
WWZZ202516 小时前
快速上手大模型:机器学习3(多元线性回归及梯度、向量化、正规方程)
人工智能·算法·机器学习·机器人·slam·具身感知
东方佑17 小时前
从字符串中提取重复子串的Python算法解析
windows·python·算法
西阳未落17 小时前
LeetCode——二分(进阶)
算法·leetcode·职场和发展
通信小呆呆17 小时前
以矩阵视角统一理解:外积、Kronecker 积与 Khatri–Rao 积(含MATLAB可视化)
线性代数·算法·matlab·矩阵·信号处理
CoderCodingNo18 小时前
【GESP】C++四级真题 luogu-B4068 [GESP202412 四级] Recamán
开发语言·c++·算法
一个不知名程序员www18 小时前
算法学习入门---双指针(C++)
c++·算法