前缀和算法习题篇(上)

1.一维前缀和

题目描述:

解法一:暴力解法:模拟

时间复杂度是O(n*q),会超时。

解法二:前缀和解法:快速求出数组中某一个连续区间的和

快速是指O(1),前缀和思想可把时间复杂度可降到O(q)。

算法思路:
  1. 先预处理出来一个前缀和数组:O(n)
  • dp[i] 表示: [1, i] 区间内所有元素的和,那么 dp[i - 1] 里面存的就是 [1, i - 1] 区间内所有元素的和。
  • 递推公式: dp[i] = dp[i - 1] + arr[i] ;
  1. 使用前缀和数组,快速求出某一个区间内所有元素的和:O(q)
  • 当询问的区间是 [l, r] 时:区间内所有元素的和为: dp[r] - dp[l - 1] 。

时间复杂度为O(q)+O(n).

为什么我们的下标要从1开始计数呢?

举例:当访问的区间是[0,2]时,区间内所有元素的和为dp[2]-dp[-1],这里的dp[-1]越界。而当访问的区间是[1,2]时,区间内所有元素的和为dp[2]-dp[0],使dp[0]=0即可,不会越界。

  • 为了处理边界情况
  • 初始化:添加虚拟结点(辅助结点)
代码实现:
cpp 复制代码
#include <iostream>
#include<vector>
using namespace std;

int main() 
{
   //1.读入数据
   int n,q;
   cin>>n>>q;
   vector<int> arr(n+1);
   for(int i=1;i<=n;i++) cin>>arr[i];

   //2.预处理出来一个前缀和数组
   vector<long long> dp(n+1);//防止溢出
   for(int i=1;i<=n;i++) dp[i]=dp[i-1]+arr[i];

   //3.使用前缀和数组
   int l=0,r=0;
   while (q--) 
   {
        cin>>l>>r;
        cout<<dp[r]-dp[l-1]<<endl;
   }
   return 0;
}

2.二维前缀和

题目描述:


解法一:暴力解法:模拟

时间复杂度是O(n* m *q),会超时。

解法二:前缀和解法

算法思路:

1.预处理出来一个前缀和矩阵: O(m*n)

  • dp[i][j]表示:从[1,1]位置到[i,j]位置,这段区间里面所有元素的和。

2.使用前缀和矩阵 : O(q)

时间复杂度为O(m*n)+O(q).

代码实现:
cpp 复制代码
#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;         
}

最后,本篇文章到此结束,感觉不错的友友们可以一键三连支持一下笔者,有任何问题欢迎在评论区留言哦~

相关推荐
ChoSeitaku28 分钟前
链表循环及差集相关算法题|判断循环双链表是否对称|两循环单链表合并成循环链表|使双向循环链表有序|单循环链表改双向循环链表|两链表的差集(C)
c语言·算法·链表
娅娅梨30 分钟前
C++ 错题本--not found for architecture x86_64 问题
开发语言·c++
DdddJMs__13533 分钟前
C语言 | Leetcode C语言题解之第557题反转字符串中的单词III
c语言·leetcode·题解
兵哥工控34 分钟前
MFC工控项目实例二十九主对话框调用子对话框设定参数值
c++·mfc
Fuxiao___37 分钟前
不使用递归的决策树生成算法
算法
我爱工作&工作love我42 分钟前
1435:【例题3】曲线 一本通 代替三分
c++·算法
娃娃丢没有坏心思1 小时前
C++20 概念与约束(2)—— 初识概念与约束
c语言·c++·现代c++
lexusv8ls600h1 小时前
探索 C++20:C++ 的新纪元
c++·c++20
lexusv8ls600h1 小时前
C++20 中最优雅的那个小特性 - Ranges
c++·c++20
白-胖-子1 小时前
【蓝桥等考C++真题】蓝桥杯等级考试C++组第13级L13真题原题(含答案)-统计数字
开发语言·c++·算法·蓝桥杯·等考·13级