[프로그래머스/JavaScript] Lv.2 연속 부분 수열 합의 개수

 

 

해당 문제는 Set을 활용한 Greedy 문제이다.

 

 

 

 

프로그래머스

 

 

 

 

---

1. 문제 설명

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

수열

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

제한사항

 

 

 

 

 

---

2. 입출력 예

입출력 예

 

 

 

 

---

3. 문제 풀이

function solution(elements) {
    // 1) 수열의 길이
    const len = elements.length;
    
    // 2) 중복 값을 방지를 위해 Set 객체 생성
    const set = new Set();
    
    // 3) 수열의 길이 만큼 반복한다.
    for(let i=1; i<=len; i++) {
    
        // 4) 수열의 길이 만큼 반복하며, i의 길이만큼 자른다.(부분 수열)
        elements.forEach((num, j) => {
        
            /* 
            5)  자르는 길이가 수열의 끝가지 도달할 경우, 
               남은 값 만큼 앞에서 자르기 위해 나머지 값을 구한다.
            */
            let re = (j+i) % len;
            
            // 6) 자르는 길이가 수열의 끝을 넘을 경우
            if((j+i)/len > 1) {
                // 7) 넘는 만큼 앞에서 잘라(0, re) 추가하고, 해당 배열의 합을 Set에 담는다.
                set.add([...elements.slice(0, re), ...elements.slice(j, j+i)].reduce((pv,cv) => pv+cv,0));
            } else {
                // 8) 넘지 않는다면, 자른 만큼(부분 수열) 합을 구해 Set에 답는다
                set.add(elements.slice(j, j+i).reduce((pv,cv) => pv+cv, 0));
            }
        });
    }
    return set.size;
}