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

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

 

정답 설명

특정 배열에서 조건을 만족하는 연속된 부분 수열을 탐색하는 경우 투 포인터로의 접근을 고려해 보면 좋습니다.

이번 문제에서 역시 부분 수열의 합이 k인 연속된 부분 수열을 탐색하는 문제이기에 투 포인터로의 접근을 고려해 보기 좋은 문제입니다.

그리고, 선택한 부분 수열의 검증 단계에서 연속된 수열의 합이 필요하므로, 누적합에 대한 아이디어가 있다면 문제 해결에 도움이 될 것입니다.

다만, 이번에는 누적합 배열을 사용할 필요는 없습니다.

 

우리에게 필요한 값은 부분 수열의 시작 인덱스, 마지막 인덱스 값입니다. 기존 배열을 탐색하며 이 두 값을 계속해서 갱신합니다. 기존 배열의 탐색 과정은 다음과 같습니다.

1. 이번 단계에서 탐색할 위치의 인덱스(i)를 구합니다.
2. i번째 인덱스의 값을 기존 누적합에 더해 줍니다.
   1. 누적합의 값이 k 보다 커지면 기억하고 있던 부분 수열의 시작 인덱스를 늘려가며 누적합에서 제외되는 위치의 값을 뺍니다.
3. 누적합의 값이 k와 같아지면 해당 부분 배열의 시작, 끝 인덱스 값을 갱신할지 선택합니다.
   1. 이전에 구했던 부분 수열의 길이보다 짧은 부분 수열의 경우에 인덱스를 갱신합니다.
4. 계속해서 부분 수열의 끝 인덱스 값을 늘려가며 탐색을 진행합니다.

 

소스 코드

class Solution {
    public int[] solution(int[] sequence, int k) {
    
        int answerStart = -1;
        int answerEnd = -1;
        int start = -1;
        int end = -1;
        int[] answerArray = new int[2];
        
        int sum = 0;
        
        while(end < sequence.length - 1){
            end++;
            sum += sequence[end];
            
            while(sum > k){
                start++;
                sum -= sequence[start];
            }
            
            if(sum == k){
                if(answerEnd == -1){
                    answerStart = start;
                    answerEnd = end;
                }
                else{
                    if(end - start < answerEnd - answerStart){
                        answerStart = start;
                        answerEnd = end;
                    }
                }
            }
        }
        
        answerArray[0] = answerStart + 1;
        answerArray[1] = answerEnd;
        
        return answerArray;
    }
}