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

문제

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

프로그래머스

 

문제 설명

 

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

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

로직 자체는 맞은 것 같으나 시간 초과&런타임 에러(재귀가 너무 깊어 오버플로우가 나는 듯 하다)로 실패.

import java.util.*;
class Solution {
    static int N, L; //k, 배열 길이
    static int maxLen = Integer.MAX_VALUE; //부분수열 길이
    static int resIdx = -1; //시작 인덱스
    static int[] arr;
    
    public int[] solution(int[] sequence, int k) {
        N = k;
        L = sequence.length;
        arr = sequence;
        
        //시작 인덱스
        for(int i = 0; i < L; i++) {
            solve(i, 1, sequence[i]);
        }
        
        //정산
        int[] answer = {resIdx, resIdx + maxLen - 1};
        return answer;
    }
    
    public void solve(int startIdx, int length, int sum) {
        //맞췄을 경우
        if(sum == N) {
            //1. 현재 최소값과 비교
            if(length < maxLen) {
                maxLen = length;
                resIdx = startIdx;
                return;
            }
        }
        if(sum > N || (startIdx + length) >= L) return; //넘김 (값 초과 or 인덱스 초과)
   
        solve(startIdx, length + 1, sum + arr[startIdx + length]);
    }
}

 

2차 풀이(성공)

 

이 문제는 투 포인터(슬라이딩 윈도우)를 사용해야 한다.

<비내림차순 수열>

연속된 부분 수열의 합을 증가시키려면 오른쪽으로 이동해야 하며, 합을 감소시키려면 왼쪽으로 이동하면 된다.

 

for (int end = 0, start = 0; end < sequence.length; end++) {
    sum += sequence[end]; // 부분 수열의 합에 sequence[end]를 더한다.

    while (sum > k) { // 부분 수열의 합이 k보다 크면,
        sum -= sequence[start++]; // 부분 수열의 시작 부분을 하나 줄이고(sum에서 빼준다), start 포인터를 오른쪽으로 이동한다.
    }

    if (sum == k) { // 부분 수열의 합이 k와 같으면,
        // 현재 부분 수열의 길이와 시작 인덱스를 기록하거나 업데이트한다.
    }
}

 

class Solution {
    public int[] solution(int[] sequence, int k) {
        int sum = 0;

        int minLength = Integer.MAX_VALUE;
        int startIndex = 0;
        int[] answer = {-1, -1};
        
        for(int start = 0, end = 0; end < sequence.length; end++) {
            sum += sequence[end]; //오른쪽으로 이동하며 값 증가
            
            while(sum > k) { //증가한 값이 커졌다면 작아질 때까지 start를 줄이기
                sum -= sequence[start++];
            }
            
            if(sum == k) { //조건이 맞을 때
                if(end - start < minLength) { //더 짧은 길이
                    minLength = end - start;
                    startIndex = start;
                }
            }
        }
        
        answer[0] = startIndex;
        answer[1] = startIndex + minLength;
        return answer;
    }
}