[프로그래머스 Level.2] 연속된 부분 수열의 합 (연습문제) (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의 부분 수열로 만들 수 있는 값입니다.

 

---

 

입출력 예

 

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

 

 

 

 

 

 

나의 코드

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = new int[2];
        int left = 0;
        int right = 0;
        int sum = sequence[0];
        
        // 답이 무조건 있으므로 처음에 제일 크게 초기화
        answer[0] = 0;
        answer[1] = sequence.length - 1;
        
        while(left < sequence.length && right < sequence.length && left <= right) {
            if(sum == k && right - left < answer[1] - answer[0]) {
                answer[0] = left;
                answer[1] = right;
            }
            
            if(sum < k && right < sequence.length - 1) {
                sum += sequence[++right];
            } else {
                sum -= sequence[left++];
            }
        }
        
        return answer;
    }
}

 

풀이

1. 이중 for문을 이용하면 시간 초과가 발생하기 때문에 투 포인터를 이용해서 풀어야 한다.
2. 우선 비내림차순은 오름차순을 나타내는 듯 하다....
3. 투 포인터를 나타낼 left 변수와 right 변수를 0으로 초기화 한다.
4. 합계를 나타낼 sum 변수를 sequence[0]으로 초기화 한다.
5. 답이 무조건 있기 때문에 길이가 짧은 수열이 나오면 답으로 업데이트 해나가기 위해서 answer를 가장 긴 길이인 answer[0]을 0으로 answer[1]을 sequence.length - 1으로 초기화 한다.
6. 이제 while문을 생성하고 left와 right를 움직이며 sum을 조정하며 답을 구한다.
7. left와 right는 sequence의 인덱스를 나타내므로 left와 right가 sequence.length보다 작고 left가 right 이하일 때만 while문을 반복한다.
8. sum이 k와 같다면 길이가 짧은 수열이면 업데이트 하기 위해 right - left가 answer[1] - answer[0] 보다 작으면 answer를 업데이트 한다.
9. 그리고 sum이 k보다 작고 right가 sequence.length - 1보다 작으면 k에 다가가기 위해서 right를 1 증가시키고 sequence[right]를 sum에 더해주고 sum이 k 이상이라면 sum에서 sequence[left]를 빼주고 left를 1 증가시킨다.
10. 이렇게 sum을 k로 만들기 위해 right 변수를 증가시키며 sum에 더해주고 k 이상이 되면 left변수를 증가시키며 sum에서 빼주면서 left와 right 두 포인터를 이동시키며 sum이 k가 될 때마다 answer와 비교하여 길이가 짧으면 answer에 업데이트 해주면 된다.