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

U_U·2024년 3월 6일

문제 - 연속된 부분 수열의 합

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의 부분 수열로 만들 수 있는 값입니다.

입출력 예

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]를 반환합니다.

접근 방식

시작점과 종점 인덱스를 활용해서 부분합을 더하고 빼며 주어진 값과 동일한지 확인하는 방식

이 문제는 대표적인 투포인터 문제로, 종료 조건을 잘 선언한다면 금방 풀 수 있는 문제였다.

투포인터

투포인터란, 시작점과 종점을 선언하여서 부분합을 계산해 나가는 것을 말한다. 나는 해당 문제를 아래와 같은 규칙으로 문제를 풀어나갔다.

부분합이 주어진 값보다 작을 경우 : end를 늘려서 부분합에 더한다.
부분합이 주어진 값보다 클경우 : 부분합에서 시작점을 빼고 start 카운트를 올려 탐색을 이어나간다.

종료조건

내가 문제를 풀면서 10분정도 헤맸던 것은 종료 조건이다.

while(start < sequence.length && end < sequence.length)

처음 내가 작성했던 종료조건은 단순히 start와 end가 sequence의 길이만큼이라고 정했다. 하지만 이렇게 했을 때, 내가 end++;을 먼저하고 sequence[end]를 실행하기 때문에 index초과가 발생한다. 하지만 여기서 end의 종료 조건을 end<sequence.length-1 로 바꾸면 마지막 행에 아예 가지 않는 경우가 발생한다.

그래서 나는 end++을 하고 나서 end가 sequence의 길이보다 크거나 같다면 종료하게 만들었다.

정답 코드

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = {0,sequence.length-1};
        int start = 0;
        int end = 0;
        int sum =0;
        sum += sequence[start];
        while(start < sequence.length && end < sequence.length){

            if(sum < k){
                end++;
                if(end >= sequence.length) break;
                sum += sequence[end];
            }
            else if(sum > k){
                sum -= sequence[start];
                start++;
                
            }
            else{
                if(answer[1]-answer[0]> end - start || (answer[1]-answer[0] == end - start && start < answer[0])){
                    answer[0] = start;
                    answer[1] = end;
                }
                sum -=sequence[start];
                start++;
            }
        }
        return answer;
    }
}