[프로그래머스] 연속된 부분 수열의 합 JAVA 문제풀이 - 그림설명 O

문제 설명

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

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

 

문제 풀이를 위한 투포인트 알고리즘 설명

이런식으로 포인터(index값을 갖고 있음) 2개를 활용해서 이동해가며 포인터값에 해당하는 배열의 수를 더할 때 자주 쓰인다.

k = 5 이므로 계속 더해본다.

이때는 right 포인터 값에 ++ 해주고, 

sum = sequence[left] + sequence[right]

이런식으로 계속 더하는 것이다.

sum = 1+1+1+2 = 5 이므로 k값 달성!

정답 후보군에 0(left값)과 3(right값)을 저장

이때는 이제 left 값을 옮겨서 다시 더해서 비교해보는 것이다.

left 포인터를 옮겼으니 기존의 값은 합계에서 빼주는 작업이 필요하다. 

이런식으로 두개의 포인터를 활용해서 문제를 푸는 것이다.

 

문제 풀이

위의 알고리즘을 조금 응용해보겠다.

1️⃣ 우선 어차피 처음은 left가 0이니까 right 포인터를 기준으로 차례대로 합계를 더해준다.

2️⃣ sum이 k와 같으면, right 포인터에서 left 포인터를 빼서 원소의 길이를 저장해두고, 이것을 기준으로 짧은 것을 답에다가 넣는다.

( 어차피 순차적으로 반복되므로, 같은 길이 일때는 자동으로 left인덱스가 더 앞선 것을 담아두게 된다 )

 

 

정답

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = new int[2];
        // 투 포인트 선언
        int left = 0;
        int right = 0;
        // 합계값
        int sum = 0;
        int size = sequence.length;
        
        for(right = 0; right<sequence.length; right++){
            //차례대로 더해줌
            sum += sequence[right];
            
            //sum이 k보다 크다면, left를 한 칸 오른쪽으로 이동
            while(right<sequence.length && sum > k){
                // 이동한 값을 빼준다
                sum -=sequence[left];
                left++;
            }
            // sum이 같으면
            if (sum == k){
                // 원소간 길이 비교 후 짧은 것을 답에다 넣는다
                if (size > right - left){
                    size = right - left;
                    answer[0] = left;
                    answer[1] = right;
                }
            }   
        }
        return answer;
    }
}