[프로그래머스] 연속 부분 수열 합의 개수

박먼지·2023년 1월 2일
0

코딩테스트

목록 보기
11/23
post-thumbnail

💡 문제

https://school.programmers.co.kr/learn/courses/30/lessons/131701

문제 설명

철호는 수열을 가지고 놀기 좋아합니다. 어느 날 철호는 어떤 자연수로 이루어진 원형 수열의 연속하는 부분 수열의 합으로 만들 수 있는 수가 모두 몇 가지인지 알아보고 싶어졌습니다. 원형 수열이란 일반적인 수열에서 처음과 끝이 연결된 형태의 수열을 말합니다. 예를 들어 수열 [7, 9, 1, 1, 4] 로 원형 수열을 만들면 다음과 같습니다.

원형 수열은 처음과 끝이 연결되어 끊기는 부분이 없기 때문에 연속하는 부분 수열도 일반적인 수열보다 많아집니다.
원형 수열의 모든 원소 elements가 순서대로 주어질 때, 원형 수열의 연속 부분 수열 합으로 만들 수 있는 수의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항

  • 3 ≤ elements의 길이 ≤ 1,000
  • 1 ≤ elements의 원소 ≤ 1,000

내 풀이

function solution(elements) {
    elements = elements.concat(elements);
    var k = 1;
    var arr = [];
    var sum = elements[0];
    
    while(k <= elements.length/2){
        var window = sum;
    	for(let i = k; i < elements.length; i++){
        	if(i-k >= 0){
        		window += elements[i] - elements[i - k];
        	}
        	arr.push(window);
    	}
        sum+=elements[k];
        k++;
    }
    return new Set(arr).size;
}

sliding window 알고리즘을 써서 풀었다.
원래 reduce로 원하는 인덱스까지 더한 sum을 구하려고 했는데 reduce함수를 잘 몰라서 구현이 안됐다..
그래도 reduce함수 쓰는거보다 그냥 원소 하나씩 더하는게 더 간단한 듯.

sliding window 알고리즘 🪟

index 0부터 k개수만큼 더한 수를 먼저 구한 다음, k 다음 요소를 더하고 맨 앞의 요소를 빼가는 식으로 구하면 된다.

profile
개발괴발

0개의 댓글