题解 | #数组求和统计#

数组求和统计

http://www.nowcoder.com/practice/efc16ce46397436a8d1a0008c52093c1

题意整理:

整理题意既:给出两个大小为n的数组a, b,统计有多少数对(l, r)满足且a数组的[l,r]区间和等于b数组的的和

方法一:暴力枚举

核心思想:

可以对a进行二重枚举,既枚举l和r,然后计算对应值是否满足条件。第二维的枚举使用变量sum统计前面的数的和,使得对单个区间计算的复杂度为1

核心代码:

class Solution {
public:
    int countLR(vector<int>& a, vector<int>& b) {
        int n = a.size();
        int ans = 0;
        for(int i = 0; i < n; ++i) {//枚举l
            int sum = 0;//隐式前缀和
            for(int j = i; j < n; ++j) {//枚举j
                sum += a[j];
                if(sum == b[i] + b[j]) {//满足条件
                    ++ans;
                }
            }
        }
        return ans;
    }
};

复杂度分析:

时间复杂度:。使用了二重枚举,复杂度为
空间复杂度:,只使用了常数个变量

方法二:

核心思想:

方法一采用了隐式的前缀和,既sum变量,也可以通过显式的前缀和进行计算。前缀和数组不需要创建,可以通过直接对a进行修改完成。
前缀和
图片说明

核心代码:

class Solution {
public:
    int countLR(vector<int>& a, vector<int>& b) {
        int n = a.size();
        for(int i = 1; i < n; ++i) {
            a[i] += a[i - 1];//将其改为前缀和数组
        }
        int ans = 0;
        for(int i = 0; i < n; ++i) {
            for(int j = i; j < n; ++j) {
                //计算当前和
                int p = i == 0 ? a[j] : a[j] - a[i - 1];//对0特判
                if(p == b[i] + b[j]) {
                    ++ans;
                }
            }
        }
        return ans;
    }
};

复杂度分析:

时间复杂度:。使用了二重枚举,复杂度为
空间复杂度:,只使用了常数个变量

方法三:

核心思想:

方法二使用了前缀和,可是因为二重枚举使得时间复杂度仍为。考虑能否对其优化
观察方法二代码,发现实际上是在计算满足 的(l,r)对数量
改写上述等式为,等式左侧和右侧都只有一个数字,所以可以进行一维的枚举实现。
更具体的说,在计算出a的前缀数组sum后,既计算 ,
可以通过一个map记录之前的每个完成统计。

核心代码:

class Solution {
public:
    int countLR(vector<int>& a, vector<int>& b) {
        int n = a.size();
        int ans = 0;
        vector<int> sum(n);//前缀数组
        sum[0] = a[0];
        //计算前缀数组
        for(int i = 1; i < n; ++i) {
            sum[i] = sum[i - 1] + a[i];
        }
        unordered_map<int, int> mp;//统计之前的sum[l] - a[l] + b[l]
        for(int i = 0; i < n; ++i) {
            int p = sum[i] - a[i] + b[i];
            mp[p]++;//进行l统计
            if(mp.find(sum[i] - b[i]) != mp.end()) {
                //进行r求和
                ans += mp[sum[i] - b[i]];
            }
        }
        return ans;
    }
};

复杂度分析:

时间复杂度:。前缀和数组的计算为。统计时只进行一维枚举,对unordered_map单次操作的时间复杂度为,故总的时间复杂度为
空间复杂度:,既前缀数组和哈希表的开销,两者均为

全部评论

相关推荐

原来已经一年了,因为没有加任何实验室没有学长学姐带,再一次偶然的机会下刷到我们学校的牛肉哥,和他聊天之后发现他也没加实验室能进大厂,我就燃起了希望,去年大概&nbsp;4&nbsp;月份找好路线&nbsp;零基础&nbsp;开始学&nbsp;5&nbsp;月背八股和开始刷算法很难受&nbsp;7-8&nbsp;月焦虑躯体化害怕找不到实习&nbsp;9&nbsp;月找到一家像样的小厂去实习了&nbsp;4&nbsp;个月大三上期末考试结束之后&nbsp;1&nbsp;月份回来边实习边准备工作压力很大&nbsp;当时只有字节、百度、商汤的面试,字节三面挂了,百度&nbsp;oc,商汤&nbsp;二面挂(差评&nbsp;无效面试),之后来深圳百度实习之后还是觉得不甘心一直没把算法和八股扔下一直在准备,百度实习的时候&nbsp;mt&nbsp;交给我一个特别重要的工作数据库迁移(特别感谢&nbsp;mt&nbsp;,这个需求学到了很多东西处理了一堆线上问题),本来看着暑期他们面试都很困难,然后听说百度要涨实习薪资(然而&nbsp;5&nbsp;月并没有涨),就想着留在百度吧也懒得面试了,4&nbsp;月&nbsp;20&nbsp;多的时候字节&nbsp;hr&nbsp;打电话约面问我要不要尝试一下询问了&nbsp;1&nbsp;月份三面为啥会挂有没有学习&nbsp;ai&nbsp;知识(因为字节这边后端岗位偏&nbsp;ai),我来到百度之后全面拥抱&nbsp;AI&nbsp;也认识了我的好兄弟&nbsp;X&nbsp;哥,他在百度&nbsp;XX&nbsp;部门&nbsp;Agent&nbsp;实习,他属于是我&nbsp;Agent&nbsp;的启蒙老师,来百度之后一直在了解&nbsp;AI&nbsp;这一块,我就接受了字节的面试,一面的时候&nbsp;20&nbsp;分钟实习拷打然后突然说&nbsp;30&nbsp;分钟代码考核我心就凉了以为是&nbsp;kpi,算法题是手撕高并发安全下的令牌桶限流器,我写了整整&nbsp;80&nbsp;多行代码最后也写出来了,但是从来没看到过出这种题能&nbsp;oc&nbsp;的我也就不管了,后边面试也是很顺利但是流程有点长可能一直在横向吧总结结果是好的!!!感谢这一年努力的自己和遇到的各位互联网大佬分享的知识!!!ps&nbsp;图二纯感慨&nbsp;(觉得🍬请不要喷我)欢迎大家一起交流学习呀!!!!
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务