⭐ Lv. 2 - 연속 부분 수열 합의 개수
철호는 수열을 가지고 놀기 좋아합니다. 어느 날 철호는 어떤 자연수로 이루어진 원형 수열의 연속하는 부분 수열의 합으로 만들 수 있는 수가 모두 몇 가지인지 알아보고 싶어졌습니다. 원형 수열이란 일반적인 수열에서 처음과 끝이 연결된 형태의 수열을 말합니다. 예를 들어 수열 [7, 9, 1, 1, 4] 로 원형 수열을 만들면 다음과 같습니다.
원형 수열은 처음과 끝이 연결되어 끊기는 부분이 없기 때문에 연속하는 부분 수열도 일반적인 수열보다 많아집니다.
원형 수열의 모든 원소 elements
가 순서대로 주어질 때, 원형 수열의 연속 부분 수열 합으로 만들 수 있는 수의 개수를 return 하도록 solution 함수를 완성해주세요.
elements
의 길이 ≤ 1,000elements
의 원소 ≤ 1,000elements | result |
---|---|
[7,9,1,1,4] | 18 |
입출력 예 #1
길이가 1인 연속 부분 수열로부터 [1, 4, 7, 9] 네 가지의 합이 나올 수 있습니다.
길이가 2인 연속 부분 수열로부터 [2, 5, 10, 11, 16] 다섯 가지의 합이 나올 수 있습니다.
길이가 3인 연속 부분 수열로부터 [6, 11, 12, 17, 20] 다섯 가지의 합이 나올 수 있습니다.
길이가 4인 연속 부분 수열로부터 [13, 15, 18, 21] 네 가지의 합이 나올 수 있습니다.
길이가 5인 연속 부분 수열로부터 [22] 한 가지의 합이 나올 수 있습니다.
이들 중 중복되는 값을 제외하면 다음과 같은 18가지의 수들을 얻습니다.
[1, 2, 4, 5, 6, 7, 9, 10, 11, 12, 13, 15, 16, 17, 18, 20, 21, 22]
원형 큐와 관련된 문제입니다. 원형 큐에서의 모든 연속 부분 수열에 대한 합을 구하기 위해 다음을 반복하였습니다.
- 원형 큐의 첫 index부터 마지막까지 순회하며 다음을 반복합니다.
- 현 index부터 원형 큐를 한 바퀴 돌아 자신 전까지 순차적으로 해당 요소를 누적하여 더한 값을 매번 배열에push
합니다.set
을 이용하여 중복된 요소를 제거한 이후, 해당 배열의 길이를 return 합니다.
function solution(e) {
let result = [];
const len = e.length;
for(let i = 0; i < len; i++) {
let current = 0;
for(let j = 1; j <= len; j++) {
current += e[(len + i + j) % len];
result.push(current);
}
}
return [...new Set(result)].length;
}