【Leetcode】1588. Sum of All Odd Length Subarrays

最新推荐文章于 2024-10-16 10:13:45 发布

记录算法题解

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量271

点赞数

分类专栏： LC 栈、队列、串及其他数据结构文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/qq_46105170/article/details/112705660

版权

LC 栈、队列、串及其他数据结构专栏收录该内容

716 篇文章 8 订阅

订阅专栏

这篇博客介绍了如何利用前缀和在O(n^2)的时间复杂度和O(n)的空间复杂度下解决LeetCode中的一个题目——计算所有奇数长度子数组的和。代码中展示了如何通过双层循环实现这一目标，从而有效地查询子数组的和。

摘要由CSDN通过智能技术生成

题目地址：

https://leetcode.com/problems/sum-of-all-odd-length-subarrays/

给定一个长 $n$ 的数组 $A$ ，求其所有长奇数的子数组的和的总和。

先算前缀和，这样可以 $O (1)$ 询问子数组和了。接着两层循环枚举即可，外层枚举子数组长度，内层枚举左端点。代码如下：

public class Solution {
    public int sumOddLengthSubarrays(int[] arr) {
        int[] preSum = new int[arr.length + 1];
        for (int i = 0; i < arr.length; i++) {
            preSum[i + 1] = preSum[i] + arr[i];
        }
        
        int res = 0;
        for (int i = 1; i <= arr.length; i += 2) {
        	// 右端点要小于arr.length
            for (int j = 0; j + i - 1 < arr.length; j++) {
                res += preSum[j + i] - preSum[j];
            }
        }
        
        return res;
    }
}