JAVASCRIPT, PROGRAMMERS, CODINGTEST, LV2

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

Apr 20, 2023 LeeJam

문제 출처

문제 설명

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

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

풀이

function solution(sequence, k) {
  // 두 인덱스 사이에 답이 있기에 기본 값으로 0 부터 끝인 1,000,000으로 세팅
  const answer = [0, 1000000];

  // left, right를 0으로 세팅
  let [left, right] = [0, 0];

  // sum에 첫 번째 값을 넣어준다.
  let sum = sequence[0];

  // right가 sequence의 길이보다 작으면 계속 루프를 돈다.
  while (right < sequence.length) {
    if (sum === k) {
      // sum이 k와 같아지면
      // 현재 답에 구해진 인덱스 사이의 길이와 새로 들어온 길이를 비교해
      // 더 짧은 길이를 넣어준다.
      if (answer[1] - answer[0] > right - left) {
        answer[0] = left;
        answer[1] = right;
      }
      // 그 후 left와 right 값을 하나씩 올려주며 처리해준다.
      sum -= sequence[left++];
      sum += sequence[++right];
    }

    // sum이 k보다 크면 sum에서 현재 left 인덱스에 해당하는 값을 빼고 left 증가
    else if (sum > k) sum -= sequence[left++];
    // sum이 k보다 작으면 right값을 증가 시키고 right 인덱스에 해당하는 값을 더해준다.
    else if (sum < k) sum += sequence[++right];
  }

  return answer;
}

정리

투 포인터 알고리즘을 사용해 문제를 풀어보았습니다.

피드백은 언제나 환영입니다. 😊