본문 바로가기

Programmers

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

https://school.programmers.co.kr/learn/courses/30/lessons/178870

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

  • 기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
  • 부분 수열의 합은 k입니다.
  • 합이 k인 부분 수열이 여러 개인 경우 길이가 짧은 수열을 찾습니다.
  • 길이가 짧은 수열이 여러 개인 경우 앞쪽(시작 인덱스가 작은)에 나오는 수열을 찾습니다.

제한사항 :

  • 5 ≤ sequence의 길이 ≤ 1,000,000
    • 1 ≤ sequence의 원소 ≤ 1,000
    • sequence는 비내림차순으로 정렬되어 있습니다.
  • 5 ≤ k ≤ 1,000,000,000
    • k는 항상 sequence의 부분 수열로 만들 수 있는 값입니다.

제한사항에 비내림차순은... 오름차순을 말하는 거겠지?

left 와 right를 두고 sum을 바꾸어나가다가 sum = k가 되었을 때

distance(right - left)가 작은 경우 정담으로 갱신

import java.util.ArrayList;
import java.util.Arrays;

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = new int[2];
        int sum = 0, left = 0, right = 0;
        int length = Integer.MAX_VALUE;
        while (right != sequence.length+1) {
            if (sum > k) {
                sum -= sequence[left];
                left++;
            } else {
                if (right == sequence.length) break;
                sum += sequence[right];
                right++;
            }
            if (sum == k) {
                int distance = right - left;
                if (length > distance) {
                    length = distance;
                    answer[0] = left;
                    answer[1] = right - 1;
                }
            }
        }
        return answer;
    }
}