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

https://school.programmers.co.kr/learn/courses/30/lessons/178870?language=javascript

프로그래머스

 

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

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

 

처음에는 DP로 접근해야하나? 라고 생각했는데 문제를 풀 수는 있겠지만 반복문을 두 번은 돌려야된다는 생각에 머리를 싸멨다.

 

예전에 백준 문제를 푼 경험으로 천천히 생각해보니 투 포인터라는 결론에 도달했다.

 

function solution(sequence, k) {
  // 초기값 설정: 구간의 시작과 끝을 저장할 변수 (길이를 최소로 설정)
  let result = [-1, -1];
  
  let left = 0; // 왼쪽 포인터
  let currentSum = 0; // 현재 구간의 합
  
  // 오른쪽 포인터를 사용하여 배열을 탐색
  for (let right = 0; right < sequence.length; right++) {
    // 오른쪽 포인터가 가리키는 값을 구간에 더한다.
    currentSum += sequence[right];

    // 구간의 합이 k보다 커지면 왼쪽 포인터를 이동시켜 구간의 합을 줄인다.
    while (currentSum > k && left <= right) {
      currentSum -= sequence[left++];
    }

    // 구간의 합이 k와 같으면, 더 짧은 구간을 찾는다.
    if (currentSum === k) {
      if (result[0] === -1 || right - left < result[1] - result[0]) {
        result = [left, right];
      }
    }
  }

  return result;
}

 

위는 정답 코드이고 아래는 내가 다시 봤을때 헤매지 않기 위해 기록한다.

 

 

if (currentSum === k) {
  if (result[0] === -1 || right - left < result[1] - result[0]) {
    result = [left, right];
  }
}

 

1. currentSum === k 조건

이 조건은 현재 구간의 합이 k와 정확히 일치할 때를 체크합니다. 구간 합이 k일 경우, 해당 구간을 결과로 기록해야 하므로, 이 조건에서부터 구간을 후속 처리하게 됩니다.

2. result[0] === -1 조건

result 배열은 구간의 시작 인덱스와 끝 인덱스를 저장하는 변수입니다. 처음에 result는 [-1, -1]로 설정되어 있습니다. 이 값은 아직 유효한 구간이 발견되지 않았음을 나타냅니다. 즉, 구간이 처음 발견되었을 때, result[0]이 -1인지를 체크하는 것입니다.

• result[0] === -1 조건은 처음으로 구간을 찾은 경우입니다.
  • 구간이 처음 발견되었을 때는 그 구간을 바로 result에 저장해야 하므로, result[0]이 -1인지를 체크합니다.

3. right - left < result[1] - result[0] 조건

이 조건은 현재 구간의 길이가 이전에 저장된 구간보다 더 짧으면 구간을 갱신하자는 의미입니다.

• right - left는 현재 구간의 길이를 의미합니다.
  • right는 현재 구간의 끝 인덱스를 나타내고, left는 현재 구간의 시작 인덱스를 나타내므로, right - left는 구간의 길이를 계산합니다.
• result[1] - result[0]는 이전에 저장된 구간의 길이를 나타냅니다.
  • result[0]은 시작 인덱스이고, result[1]은 끝 인덱스이므로, 두 값을 빼면 구간의 길이가 나옵니다.

따라서 right - left < result[1] - result[0] 조건은 현재 구간의 길이가 기존 구간보다 짧으면 구간을 갱신하겠다는 뜻입니다. 구간의 길이가 더 짧을 때 더 짧은 구간을 선택하여 result에 저장합니다.

4. result = [left, right]

위 조건들이 만족되면, 새로운 구간을 result에 기록합니다. left는 현재 구간의 시작 인덱스이고, right는 현재 구간의 끝 인덱스이므로, result 배열에 [left, right]를 저장합니다.

 

투포인터 문제 확인 방법

1. 배열/리스트 관련 문제

• 배열이나 리스트를 다루는 문제에서 종종 투 포인터를 사용할 수 있습니다.
• 정렬된 배열을 다루는 문제가 주어졌다면, 투 포인터 기법을 고려해보세요. 특히, 두 수의 합을 찾는 문제나 부분합 문제는 매우 흔한 예입니다.

2. 두 수 또는 여러 수의 합 문제

• 문제에서 "두 수의 합" 또는 "세 수의 합"을 찾는 문제일 때, 투 포인터를 사용하면 효율적으로 해결할 수 있습니다.
  • 예: 배열에서 두 수의 합이 특정 값이 되는 경우, 세 수의 합이 특정 값이 되는 경우 등.
  • 정렬된 배열에서 시작과 끝을 가리키는 두 포인터를 이용하여 합을 맞추는 방식입니다.

3. 부분합, 연속된 부분 배열

• 주어진 배열에서 연속된 부분 배열의 합이 특정 값이 되는지, 또는 그 구간을 찾는 문제도 투 포인터를 사용합니다.
• 예: "부분 배열의 합이 특정 값이 되는 경우", "연속된 부분 배열의 합을 구하는 문제" 등.
• **슬라이딩 윈도우(Sliding Window)**와 관련된 문제들도 투 포인터를 활용할 수 있습니다.

4. 정렬된 배열에서의 쌍 찾기

• 배열이 이미 정렬되어 있고, 배열 내에서 두 수의 합이나 두 수의 차가 특정 값이 되는 쌍을 찾는 문제에서 자주 쓰입니다.
  • 예: "배열에서 두 수의 합이 특정 값이 되는 쌍을 찾는 문제"

5. 배열을 두 부분으로 나누는 문제

• 배열을 두 부분으로 나누어 각각의 구간을 다루는 문제에서 투 포인터를 사용하기도 합니다.
• 예: 배열의 왼쪽과 오른쪽 구간에 대해 조건을 만족하는 쌍을 찾는 문제

6. 두 개 이상의 포인터를 사용하여 특정 조건을 만족하는 문제

• 문제에서 "두 개 이상의 포인터"가 등장하거나 구간의 크기를 조절하는 방식이 필요한 문제에서 투 포인터 기법을 생각해 볼 수 있습니다.

7.  빠른 탐색이 필요한 문제

• 문제에서 탐색을 빠르게 하도록 요구하는 경우, 투 포인터를 사용하여 배열을 한 번만 순회하는 방식으로 해결할 수 있습니다. 예를 들어, 두 수의 합을 찾을 때, 일일이 모든 가능한 쌍을 비교하는 대신, 두 포인터를 사용하여 더 효율적으로 해결할 수 있습니다.