kakasoo

[node.js] 연속 부분 수열 합의 개수 본문

프로그래밍/알고리즘 풀이

[node.js] 연속 부분 수열 합의 개수

카카수(kakasoo) 2023. 6. 4. 00:35
반응형
/**
 * 프로그래머스 연속 부분 수열 합의 개수
 * 
 * 푼 날짜 : 2023.06.04
 * 
 * @param {number[]} elements
 * @returns {number}
 */
function solution(elements) {
    /**
     * 원형으로 쓰인 수열이기 때문에 반복을 표현한다.
     * 이렇게 하지 않을 경우, `array.prototype.at()`을 이용하는 방법도 있다.
     */
    elements.push(...elements);

    /**
     * answer에 시작 지점으로부터 n개를 담았을 때의 부분 수열 합을 모두 담는다.
     * 반복되었기 때문에 2번 씩 담기는 것을 막고자 length와 start는 모두 전체 길이의 반까지만을 허용한다.
     */
    const answer = [];
    for (let length = 1; length <= parseInt(elements.length / 2); length++) {
        for (let start = 0; start < parseInt(elements.length / 2); start++) {
            if (start + length <= elements.length) {
                const sum = elements
                    .slice(start, start + length)
                    .reduce((acc, cur) => acc + cur, 0);
                answer.push(sum);
            }
        }
    }

    return Array.from(new Set(answer)).length;
}
반응형