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

문제 설명

비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.
기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 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. k보다 작은 원소까지만 탐색한다 
//2. 투포인터 이용해서 앞+뒤로 따로 탐색 값이 커지면 뒤를 당기고 값이 작아지면 앞을 민다
//3. 백준 회전초밥 문제와 동일

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = {};
        
        int min_left = 0;
        int min_right = sequence.length;
        
        int left = 0;
        int right = 0;
        int sum = sequence[0];
        
        while(left < sequence.length && right < sequence.length) {
            
            if (sequence[left] > k) {
                break;
            }
            
            //부분 수열의 합이 k일 때는 조건에 따라 업데이트
            if (sum == k) {
                if (min_right - min_left > right-left) {
                    min_left = left;
                    min_right = right;
                } else if (min_right - min_left == right - left) {
                    if (min_left > left) {
                        min_left = left;
                        min_right = right;
                    }
                }
            }
            
            //부분 수열의 합이 k보다 작다면 오른쪽을 민다
            if (sum <= k) {
                right++;
                if (right < sequence.length) {
                    sum += sequence[right];
                }
            } 
            
            //부분 수열의 합이 k보다 크다면 왼쪽을 당긴다
            else {
                sum -= sequence[left];
                left++;
            }
        }
        
        answer = new int[]{min_left, min_right};
        return answer;
    }
}

피드백

완전 탐색으로 풀 경우 O(n^2)이 나오기 때문에 1,000,000^2은 안 됨 즉 최대 O(nlogn)으로 접근해야하는데 연속된 수열의 합은 투포인터로 접근이 가능하다 난이도 7/10 걸린시간 1시간 <다시 한 번 풀어보가>