【Leetcode】1588.所有奇数长度子数组的和

题目描述


思路

题目要求我们求解所有奇数长度数组的和。若暴力循环求解,时间复杂度过高。所以,我们可以采用前缀和优化。

如上图输入arr数组,sum[i]用于计算arr数组中前i个数的和。(在程序中,先给sum[0]赋值,等于arr[0],其余sum[i]=sum[i-1]+arr[i])

接着,我们计算需要返回的ret。我们定义两个循环变量i和j。先定i不动,j向后移动,然后每移动一个位置,就判断一下,是否i和j之间相差奇数个数,若是,则ret+=sum[j]-sum[i-1]。(包括i,所以是减去前一个位置的。)

然后i作为外层循环,后续循环如上面一样判断。

代码

cpp 复制代码
class Solution {
public:
        int sumOddLengthSubarrays(vector<int>&arr) {
        
        int length=arr.size();//获取数组中元素的个数
        int sum[length];//定义前缀和数组
        long long int ret;
        ret=0;
        sum[0]=arr[0];
        //计算前缀和
        for (int i=1;i<length;i++)
        {
            sum[i]=sum[i-1]+arr[i];
        }
        
        for (int  i=0;i<length;i++)
        {
            for (int j=i;j<length;j++)
            {
                
            if ((j-i+1)%2)//若相差奇数个数
            {
            if (i==0) ret+=sum[j];//i=0时,不分开讨论sum[i-1]数组越界。
            else
                ret+=sum[j]-sum[i-1];
            }
            
            }
        }
        return ret;
        }
};
相关推荐
开发者每周简报26 分钟前
求职市场变化
人工智能·面试·职场和发展
刚学HTML1 小时前
leetcode 05 回文字符串
算法·leetcode
AC使者2 小时前
#B1630. 数字走向4
算法
冠位观测者2 小时前
【Leetcode 每日一题】2545. 根据第 K 场考试的分数排序
数据结构·算法·leetcode
@小码农2 小时前
202411 第十六届蓝桥杯青少组 STEMA 考试真题 汇总
职场和发展·蓝桥杯
古希腊掌管学习的神2 小时前
[搜广推]王树森推荐系统笔记——曝光过滤 & Bloom Filter
算法·推荐算法
qystca2 小时前
洛谷 P1706 全排列问题 C语言
算法
古希腊掌管学习的神3 小时前
[LeetCode-Python版]相向双指针——611. 有效三角形的个数
开发语言·python·leetcode
浊酒南街3 小时前
决策树(理论知识1)
算法·决策树·机器学习
就爱学编程3 小时前
重生之我在异世界学编程之C语言小项目:通讯录
c语言·开发语言·数据结构·算法