[LeetCode] Sum of All Odd Length Subarrays

준규·2022년 12월 18일

1.문제


Given an array of positive integers arr, return the sum of all possible odd-length subarrays of arr.

A subarray is a contiguous subsequence of the array.


정수 배열이 주어질 때 홀수 길이의 부분 배열들의 숫자들의 합을 리턴하는 문제이다.


Example 1

Input: arr = [1,4,2,5,3]
Output: 58
Explanation: The odd-length subarrays of arr and their sums are:
[1] = 1
[4] = 4
[2] = 2
[5] = 5
[3] = 3
[1,4,2] = 7
[4,2,5] = 11
[2,5,3] = 10
[1,4,2,5,3] = 15
If we add all these together we get 1 + 4 + 2 + 5 + 3 + 7 + 11 + 10 + 15 = 58

Example 2

Input: arr = [1,2]
Output: 3
Explanation: There are only 2 subarrays of odd length, [1] and [2]. Their sum is 3.

Example 3

Input: arr = [10,11,12]
Output: 66

Constraints:

  • 1 <= arr.length <= 100
  • 1 <= arr[i] <= 1000

2.풀이

  1. 배열의 총 길이 이내로 홀수개의 부분배열로 잘라야한다.
  2. 홀수개씩 부분 배열을 잘라서 부분배열 숫자들의 총합을 sum에 더해준다.

/**
 * @param {number[]} arr
 * @return {number}
 */
const sumOddLengthSubarrays = function (arr) {
  let sum = 0;
  for (let i = 1; i <= arr.length; i += 2) {
    // 홀수 개씩 subarray
    for (let j = 0; j <= arr.length - i; j++) {
      //배열 앞에서부터 홀수개씩 배열을 자른다
      let temp = arr.slice(j, i + j);
      sum += temp.reduce((sum, item) => sum + item); // 배열의 합을 sum에 더해준다
    }
  }

  return sum;
};

3.결과

profile
안녕하세요 :)

0개의 댓글