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

lee-goeun·2023년 7월 10일
0

문제출처
https://school.programmers.co.kr/learn/courses/30/lessons/178870

문제 설명

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

기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 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의 부분 수열로 만들 수 있는 값입니다.

입출력 예

문제풀이

  1. 포인터가 될 head변수와 합을 나타낼 sum변수를 선언한다. 그리고 정답을 담아줄 변수를 선언한다.
  2. sequence배열을 돌면서 하나씩 더해준다.
  3. 만약 sum이 k값보다 크면 작을때까지 빼준다.(head를 포인터삼아 하나씩 늘려준다)
  4. 만약 sum이 k와 같다면 정답에 head와 i를 넣어준다.
  5. 배열을 다 돌게되면 정답 배열에 k와 값이 같은 index 배열들이 있다.
  6. 이 배열을 돌면서 길이값이 가장 작은 값을 찾는다. 가장 작은 값이 있으면 결과변수에 넣어준다.

다른사람 코드

풀이 팁을 봐도 도저히 모르겠어서 다른 사람 코드를 봤다. 이제서야 하나씩 빼면서 더하라는 게 무슨 말인지 알겠다.

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;
}

0개의 댓글