Algorithm

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

cob 2023. 6. 30. 14:46

 

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 


1. 문제 설명

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

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

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

제한사항

 

 

 

 


2. 입출력

입출력

 

 

 

 


3. 문제 풀이

투 포인터  => 부분 수열 합과 동일할 경우 i, j 동시 이동 
function solution(sequence, k) {
    const answer = [];
    let [i, j] = [0, 0];        // 투 포인터 설정
    let sum = sequence[0];      // 0번 인덱스 값을 미리 셋팅
    let minLen = Infinity;      // 최솟값 
    let nowLen = 0;             // 현재값
    
    
    while(j < sequence.length){
        if(sum === k){
            nowLen = j - i;        
            if(minLen > nowLen){
                answer[0] = i;
                answer[1] = j;    
                minLen = nowLen;
            }
        
        	// 현재 sequence[i] 값을 빼고 i를 증가 시켜야 하므로 i++
        	sum -= sequence[i++];
            
            // 다음 sequence[j] 값을 더하고 j를 증가 시켜야 하므로 ++j
            sum += sequence[++j];
        }
        else if(sum > k) sum -= sequence[i++];
        else if(sum < k) sum += sequence[++j];
    }
    return answer;
}

 

 

반응형