C++ 数据结构与算法(前缀和/差分算法)

前缀和

https://www.jianshu.com/p/89ec2814682c
入门

前缀和是一种重要的预处理,能大大降低查询的时间复杂度。可以简单理解为“数列的前 n 项的和”。
前缀和是一个数组的某项下标之前(包括此项元素)的所有数组元素的和。
b[] 为前缀和数组,a[] 为原数组,根据这句话可以得到前缀和的定义式和递推式:
在这里插入图片描述

2055. 蜡烛之间的盘子

预处理 + 前缀和

右边界前蜡烛总数 - 左边界前蜡烛总数

  • 时间复杂度: O ( n + q ) O(n+q) O(n+q),其中 n 为数组长度,q 为询问数量。我们需要 O(n) 的时间预处理。对于每一个询问,我们需要 O(1) 的时间计算答案。
  • 空间复杂度: O ( n ) O(n) O(n),其中 n 为数组长度。我们需要 O(n) 的空间保存预处理的结果。注意返回值不计入空间复杂度
class Solution {
public:
    vector<int> platesBetweenCandles(string s, vector<vector<int>>& queries) {
        int n = s.length();
        vector<int> nums(n);		// 保存当前位置前的盘子数量
        int sum = 0;
        for(int i = 0; i < n; ++i){
            if(s[i] == '*'){
                ++sum;
            }
            nums[i] = sum;
        }

        int l = -1;
        vector<int> left(n,-1);		// 当前位置右边第一根蜡烛的位置,即查询的左边界
        for(int i = n-1; i >= 0; --i){
            if(s[i] == '|'){
                l = i;
            }
            left[i] = l;
        }

        l = -1;
        vector<int> right(n,-1);	// 当前位置左边第一根蜡烛的位置,即查询的右边界
        for(int i = 0; i < n; ++i){
            if(s[i] == '|'){
                l = i;
            }
            right[i] = l;
        }

        int nn = queries.size();
        vector<int> ans(nn,0);		
        for(int i = 0; i < nn; ++i){		// 查询,取差
            int lx = left[queries[i][0]], rx = right[queries[i][1]];
            if(lx != -1 && rx != -1 && lx < rx){	
                ans[i] = nums[rx] - nums[lx];
            }
        }
        return ans;
    }
};
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值