[Programmers] 연속된 부분 수열의 합 - JavaScript

Joosi_Cool·2023년 4월 12일

2단계 JavaScript programmers 연속된 부분 수열의 합

Programmers

목록 보기

66/98

문제설명

비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.

기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 k입니다.
합이 k인 부분 수열이 여러 개인 경우 길이가 짧은 수열을 찾습니다.
길이가 짧은 수열이 여러 개인 경우 앞쪽(시작 인덱스가 작은)에 나오는 수열을 찾습니다.

수열을 나타내는 정수 배열 sequence와 부분 수열의 합을 나타내는 정수 k가 매개변수로 주어질 때, 위 조건을 만족하는 부분 수열의 시작 인덱스와 마지막 인덱스를 배열에 담아 return 하는 solution 함수를 완성해주세요. 이때 수열의 인덱스는 0부터 시작합니다.

제한사항

5 ≤ sequence의 길이 ≤ 1,000,000
- 1 ≤ sequence의 원소 ≤ 1,000
- sequence는 비내림차순으로 정렬되어 있습니다.
5 ≤ k ≤ 1,000,000,000
- k는 항상 sequence의 부분 수열로 만들 수 있는 값입니다.

입출력 예

sequence	k	result
[1, 2, 3, 4, 5]	7	[2, 3]
[1, 1, 1, 2, 3, 4, 5]	5	[6, 6]
[2, 2, 2, 2, 2]	6	[0, 2]

입출력 예 설명

입출력 예 #1

[1, 2, 3, 4, 5]에서 합이 7인 연속된 부분 수열은 [3, 4]뿐이므로 해당 수열의 시작 인덱스인 2와 마지막 인덱스 3을 배열에 담아 [2, 3]을 반환합니다.

입출력 예 #2

[1, 1, 1, 2, 3, 4, 5]에서 합이 5인 연속된 부분 수열은 [1, 1, 1, 2], [2, 3], [5]가 있습니다. 이 중 [5]의 길이가 제일 짧으므로 해당 수열의 시작 인덱스와 마지막 인덱스를 담은 [6, 6]을 반환합니다.

입출력 예 #3

[2, 2, 2, 2, 2]에서 합이 6인 연속된 부분 수열은 [2, 2, 2]로 3가지 경우가 있는데, 길이가 짧은 수열이 여러 개인 경우 앞쪽에 나온 수열을 찾으므로 [0, 2]를 반환합니다.

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

설계 과정

우선 이 문제는 배열을 잘라서 접근하면 쉽게 답을 구할 수 있긴하다.하지만 2단계인 만큼 시간복잡도를 신경써야 할 것이다. 따라서 본인은 O(n)의 시간복잡도를 목표로 문제를 설계했다.

우선 head = 0 => 이 변수는 앞에를 자를때 쓰이는 기준 인덱스 값이다.
1. sequence를 한번씩 순회하면서 sum이라는 변수에 더해준다.
2. 이때 sum>k를 넘어가면 앞에 배열 값부터 크지 않을때까지 빼준다.
3. 이때 만약 값이 같다면 정답 가능성이 있는것으로 판단하여 answer에 추가
4. 모두 순회하고 나면 길이가 가장 짧은 것을 찾는다.

정답 코드

function solution(sequence, k) {
  //정답이 될 수 있는 값을 담을 배열
    var answer = [];
    var sum = 0;
    var head = 0;
    for(var i = 0; i<sequence.length;i++){
        sum+=sequence[i];
        if(sum>k){
          // 작아질때까지 앞에 값을 빼줌.
            while(sum>k){
                sum -= sequence[head++];
            }
        }
        //같다면 리턴
        if(sum===k){
            answer.push([head,i]);
        }
    }
    var min = sequence.length;
    var result = [];
    answer.forEach((element)=>{
        if(min>(element[1]-element[0])){
            min = (element[1]-element[0]);
            result = [element[0],element[1]];
        }
    }) 
    return result;
}