【leetcode】动态规划::前缀和

这篇具有很好参考价值的文章主要介绍了【leetcode】动态规划::前缀和。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

标题:【leetcode】前缀和

@水墨不写bug

【leetcode】动态规划::前缀和,决胜oj,leetcode,算法,动态规划,数据结构


正文开始:

(一)简单前缀和

描述

给定一个长度为n的数组a1​,a2​,....an​.

接下来有q次查询, 每次查询有两个参数l, r.

对于每个询问, 请输出al​+al+1​+....+ar​

输入描述:

第一行包含两个整数n和q.

第二行包含n个整数, 表示a1​,a2​,....an​.

接下来q行,每行包含两个整数   l和r.

1≤n,q≤10^5
-10^9≤a[i]≤10^9
1≤l≤r≤n

输出描述:

输出q行,每行代表一次查询的结果.

示例1

输入:

3 2
1 2 4
1 2
2 3

输出:

3
6

         思路一: 暴力求解,时间复杂度O(N^2),看到下方的数据量,显然是会超时的算法;

        思路二:动态规划

        动态规划是一种解决多阶段决策问题的数学优化方法。它将原问题分解为若干个子问题,并把子问题的解保存起来,避免重复计算,从而减少计算量。动态规划通常适用于具有重叠子问题和最优子结构的问题,通过构建一个递推关系式,将问题的最优解表示成子问题的最优解的组合。通过自底向上的方式,从子问题的最优解逐步推导出原问题的最优解。动态规划可以大大提高问题的求解效率,但也需要额外的空间来存储子问题的解。

        通过这一思想,我们可以创建一个dp数组:它的每一个下标i对应的位置存储和它对应下标的arr的前(i)个元素和:

        这时若要求得 [l,r] 闭区间的两个下标之间的元素之和,只需对dp数组操作即可。解题关键就是找到如何构造dp数组的方法,以及如何利用dp数组来得到结果。

        本题构建dp数组的策略比较简单:dp数组的前一项加上arr数组的本项即可得到dp数组的本项:

dp[k] = dp[k-1] + arr[k]

        利用dp数组的方法也很简单:输出dp的r下标和dp的(l - 1)下标之和即可:

cout<<(long)dp[r]-(long)dp[l-1]<<endl;

        时间复杂度O(q);

【leetcode】动态规划::前缀和,决胜oj,leetcode,算法,动态规划,数据结构

       

        考虑到题目需要做加法,并且数据量达到10 ^5,数据返回范围达到了10^9量级,所以数据类型开成long

参考代码: 

#include <iostream>
#include <vector>
using namespace std;
int main() {
    int n,q;
    cin>>n>>q;
    vector<long> arr(n+1);
    for(int i = 1;i < n+1;i++) cin>>arr[i];
    vector<long> dp(n+1);//构造数组dp
    for(int k = 1;k < n+1;k++) dp[k] = dp[k-1] + arr[k];
    for(int j = 0;j < q;j++)//主逻辑
    {
        int l = 0,r = 0;
        cin>>l>>r;
        cout<<(long)dp[r]-(long)dp[l-1]<<endl;
    }
}

(二)二维前缀和

描述

给你一个 n 行 m 列的矩阵 A ,下标从1开始。

接下来有 q 次查询,每次查询输入 4 个参数 x1 , y1 , x2 , y2

请输出以 (x1, y1) 为左上角 , (x2,y2) 为右下角的子矩阵的和,

输入描述:

第一行包含三个整数n,m,q.

接下来n行,每行m个整数,代表矩阵的元素

接下来q行,每行4个整数x1, y1, x2, y2,分别代表这次查询的参数

1≤n,m≤1000
1≤q≤10^5
−109≤a[i][j]≤10^9
1≤x1​≤x2​≤n
1≤y1​≤y2​≤m

输出描述:

输出q行,每行表示查询结果。

示例1

输入:

3 4 3
1 2 3 4
3 2 1 0
1 5 7 8
1 1 2 2
1 1 3 3
1 2 3 4

输出:

8
25
32

备注:

读入数据可能很大,请注意读写时间。

         经过了一维前缀和的铺垫,对前缀和有了一些初步理解,在解决二维前缀和就容易多了:

        题目要求是返回两个坐标之间的元素和:

【leetcode】动态规划::前缀和,决胜oj,leetcode,算法,动态规划,数据结构

思路一:暴力求解,时间复杂度O(q*N^2),显然是会超时的算法。

 思路二:用动态规划思想,采用二维前缀和方法。

        首先,我们可以定义一个辅助矩阵dp,其中dp[i][j]表示以(1,1)为左上角,(i,j)为右下角的子矩阵的和。

                                (下标为0的行和列不使用)

        接下来,我们可以通过以下方程来计算dp数组:

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

        其中,dp[i-1][j]表示以(1,1)为左上角,(i-1,j)为右下角的子矩阵的和; dp[i][j-1]表示以(1,1)为左上角,(i,j-1)为右下角的子矩阵的和; dp[i-1][j-1]表示以(1,1)为左上角,(i-1,j-1)为右下角的子矩阵的和; A[i][j]表示矩阵A的元素。

        对于每次查询(x1, y1)(x2, y2),我们可以通过如下公式计算子矩阵的和:

sum = dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1]

        其中,dp[x2][y2]表示以(1,1)为左上角,(x2,y2)为右下角的子矩阵的和; dp[x1-1][y2]表示以(1,1)为左上角,(x1-1,y2)为右下角的子矩阵的和; dp[x2][y1-1]表示以(1,1)为左上角,(x2,y1-1)为右下角的子矩阵的和; dp[x1-1][y1-1]表示以(1,1)为左上角,(x1-1,y1-1)为右下角的子矩阵的和。

        其实前缀和是一类题型,我们可以在一定程度上记忆状态转移方程(上述的dp表构建方程,使用dp表方程), 但是我认为更重要的是理解方程为什么是这样的结构,以及方程的推导由来:

         【leetcode】动态规划::前缀和,决胜oj,leetcode,算法,动态规划,数据结构【leetcode】动态规划::前缀和,决胜oj,leetcode,算法,动态规划,数据结构

参考代码: 

#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));
    for(int i = 1;i <= n ; i++)//读取数据
        for(int j = 1;j <= m ; j++)
            cin>>arr[i][j];
    vector<vector<long>> dp(n+1,vector<long>(m+1));//long防止溢出
    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];
    for(int c = 0;c<q;c++)//c仅仅起计数作用
    {
        int i0 = 0,j0 = 0,i = 0,j = 0;
        cin>>i0>>j0>>i>>j;
        long ret = dp[i][j] - dp[i][j0-1] - dp[i0-1][j]+dp[i0-1][j0-1];
        cout<<ret<<endl;
    }
}
// 64 位输出请用 printf("%lld")

(三)寻找数组的中心下标

给你一个整数数组 nums ,请计算数组的 中心下标 

数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。

如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。

如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1 。

示例 1:

输入:nums = [1, 7, 3, 6, 5, 6]
输出:3
解释:
中心下标是 3 。
左侧数之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 ,
右侧数之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ,二者相等。

示例 2:

输入:nums = [1, 2, 3]
输出:-1
解释:
数组中不存在满足此条件的中心下标。

示例 3:

输入:nums = [2, 1, -1]
输出:0
解释:
中心下标是 0 。
左侧数之和 sum = 0 ,(下标 0 左侧不存在元素),
右侧数之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。

提示:

  • 1 <= nums.length <= 10^4
  • -1000 <= nums[i] <= 1000

思路一:

        暴力求解,对于每一个随机下标 i ,对前(0 ~ i-1)项元素进行求和,再对其后(i+1 ~ n-1)项元素进行求和,如果两个和相等,那么返回下标;如果不相等,继续遍历,如果对于所有的下标都不想等,则返回-1。

        时间复杂度O(N^2)。

思路二:

        创建两个dp数组,分别记为 dp,dp_re 每个元素分别表示这个下标之前的元素和,这个元素之后的元素和,在比较时只需比较dp和dp_re对应位置的元素是否相等即可。建立dp表的时间复杂度为O(N),判断是否相等时遍历的时间复杂度为O(N),在使用dp表的时间复杂度则更低,达到了常数级O(N)。

        总体时间复杂度O(N)。

        (需要特殊处理的是:dp表的第一项为0,因为第一项下标为0,0下标之前已经没有元素了,如果进行访问会发生越界;dp_re表的最后一项为0,因为最后一项下标为n-1,n-1下标之后已经没有元素了,如果进行访问会发生越界)


class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int n = nums.size();
        vector<long> f(n), g(n);
        for(int i = 1;i < n;i++)//建f,g表
            f[i] = f[i-1] + nums[i-1];//不赋值默认为0
        for(int i = n-2;i >= 0;i--)
            g[i] = g[i+1] + nums[i+1];
        for(int k = 0;k < n;k++)
            if(f[k]==g[k]) return k;
    return -1;
    }
};

 (四)除自身外数组的乘积

        给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

        题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内。

        请 不要使用除法,且在 O(n) 时间复杂度内完成此题。

示例 1:

输入: nums = [1,2,3,4]
输出: [24,12,8,6]

示例 2:

输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

提示:

  • 2 <= nums.length <= 10^5
  • -30 <= nums[i] <= 30
  • 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内

思路一:

        暴力求解,显然是会超时的算法。

思路二:

        依然是使用dp表, 你会发现本题的思路和上一题完全一致。

        创建两个dp数组,分别记为 dp,dp_re 每个元素分别表示这个下标之前的元素积,这个元素之后的元素积,在比较时只需比较dp和dp_re对应位置的元素是否相等即可。建立dp表的时间复杂度为O(N),判断是否相等时遍历的时间复杂度为O(N),在使用dp表的时间复杂度则更低,达到了常数级O(N)。

        总体时间复杂度O(N)。

        (需要特殊处理的是:dp表的第一项手动初始化为1,因为第一项下标为0,0下标之前已经没有元素了,如果进行访问会发生越界;dp_re表的最后一项手动初始化为1,因为最后一项下标为n-1,n-1下标之后已经没有元素了,如果进行访问会发生越界) (对于乘法来说,将元素置0是没有意义的,所以将元素置1)

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) 
    {
        int n = nums.size();
        vector<int> f(n), g(n);
        f[0] = g[n-1] = 1;//乘法操作不能初始化为0
        for(int i = 1;i < n;i++)//建f,g表
            f[i] = f[i-1] * nums[i-1];//不赋值默认为0
        for(int i = n-2;i >= 0;i--)
            g[i] = g[i+1] * nums[i+1];
        vector<int> ret;
        for(int i = 0;i < n;i++)
            ret.push_back(f[i]*g[i]);
        return ret;
    }
};

完~

未经作者同意禁止转载文章来源地址https://www.toymoban.com/news/detail-847509.html

到了这里,关于【leetcode】动态规划::前缀和的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • LeetCode 1997.访问完所有房间的第一天:动态规划(DP)——4行主要代码(不需要什么前缀和)

    力扣题目链接:https://leetcode.cn/problems/first-day-where-you-have-been-in-all-the-rooms/ 你需要访问  n 个房间,房间从 0 到 n - 1 编号。同时,每一天都有一个日期编号,从 0 开始,依天数递增。你每天都会访问一个房间。 最开始的第 0 天,你访问  0 号房间。给你一个长度为 n 且 下标从

    2024年04月14日
    浏览(37)
  • OJ刷题---[算法课动态规划]走网格(C++完整代码)

    m行n列的网格,从左上角(1,1)出发,每一步只能向下或者向右,问共有多少种方法可以走到右下角(m,n); 输入参数 m n (1=m=10 1=n=10) 输出多少种走法 比如: 输入:2 3 输出:3 输入:5 5 输出:70 完整代码(C++): 结果: 注意:最后调用的时候,是调用sum(m,n),而不是sum(m+1

    2024年02月07日
    浏览(39)
  • LeetCode 2811. Check if it is Possible to Split Array【脑筋急转弯;前缀和+动态规划或记忆化DFS】中等

    本文属于「征服LeetCode」系列文章之一,这一系列正式开始于2021/08/12。由于LeetCode上部分题目有锁,本系列将至少持续到刷完所有无锁题之日为止;由于LeetCode还在不断地创建新题,本系列的终止日期可能是永远。在这一系列刷题文章中,我不仅会讲解多种解题思路及其优化,

    2024年02月13日
    浏览(32)
  • 【算法深度探索】动态规划之旅(1):挑战OJ题海,解锁15道经典难题,让你成为DP大师!

    📃 博客主页: 小镇敲码人 🚀 欢迎关注:👍点赞 👂🏽留言 😍收藏 🌏 任尔江湖满血骨,我自踏雪寻梅香。 万千浮云遮碧月,独傲天下百坚强。 男儿应有龙腾志,盖世一意转洪荒。 莫使此生无痕度,终归人间一捧黄。🍎🍎🍎 ❤️ 什么?你问我答案,少年你看,下一

    2024年04月11日
    浏览(36)
  • 数据结构与算法-动态规划

    (我猜是做的多了背的题多了就自然懂了) 迭代法一般没有通用去重方式,因为已经相当于递归去重后了 这两个问题其实是一个问题,一般直接写出的没有去重的递归法,复杂度很高,此时需要使用备忘录去重,而备忘录去重时间复杂度和使用dp数组进行迭代求解时间复杂度相同

    2024年02月04日
    浏览(43)
  • Leetcode题解-算法-动态规划(python版)

    1.1 爬楼梯 70. 爬楼梯(Easy) 走n阶楼梯的方法有两种,1、先走 1 级台阶,再走 n-1 级台阶;2、先走 2 级台阶,再走 n-2 级台阶 f(n) = f(n-1) + f(n-2) 1.2 强盗抢劫 198. 打家劫舍(Medium) 每个房间财产为 nums[0]……nums[n-1]。 假设 0 至 x 间房获得的最大财产为 f(x)。 f(x) = max(f(x-1),f(x-2)+nums[

    2024年02月03日
    浏览(47)
  • 数据结构与算法 | 动态规划算法(Dynamic Programming)

    上一篇文末已经提到了记忆化搜索是动态规划(Dynamic Programming)的一种形式,是一种自顶向下(Top-Down)的思考方式,通常采用递归的编码形式;既然动态规划有自顶向下(Top-Down)的递归形式,自然想到对应的另外一种思考方式 自底向上( Bottom-Up ) ,也就是本篇要写的内

    2024年02月05日
    浏览(44)
  • python算法与数据结构---动态规划

    记不住过去的人,注定要重蹈覆辙。 对于一个模型为n的问题,将其分解为k个规模较小的子问题(阶段),按顺序求解子问题,前一子问题的解,为后一子问题提供有用的信息。在求解任一子问题时,通过决策求得局部最优解,依次解决各子问题。最后通过简单的判断,得到

    2024年02月20日
    浏览(65)
  • 数据结构与算法之贪心&动态规划

            一:思考         1.某天早上公司领导找你解决一个问题,明天公司有N个同等级的会议需要使用同一个会议室,现在给你这个N个会议的开始和结束 时间,你怎么样安排才能使会议室最大利用?即安排最多场次的会议?电影的话 那肯定是最多加票价最高的,入场

    2024年02月09日
    浏览(47)
  • 【算法】动态规划 ⑦ ( LeetCode 55. 跳跃游戏 | 算法分析 | 代码示例 )

    LeetCode 55. 跳跃游戏 : https://leetcode.cn/problems/jump-game/ 给定一个 非负整数数组 nums ,你最初位于数组的 第一个下标 0 位置 。 数组中的每个元素 代表你在该位置可以 跳跃的最大长度。 判断你 是否能够到达最后一个下标。 给定一个一维数组 , 数组元素不能有负数 , 如 : {2, 2,

    2024年02月10日
    浏览(38)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包