[인프런] 최대 길이 연속 부분수열 - 자바

문제

0과 1로 구성된 길이가 N인 수열이 주어집니다. 여러분은 이 수열에서 최대 k번을 0을 1로 변경할 수 있습니다. 여러분이 최대 k번의 변경을 통해 이 수열에서 1로만 구성된 최대 길이의 연속부분수열을 찾는 프로그램을 작성하세요.
만약 길이가 길이가 14인 다음과 같은 수열이 주어지고 k=2라면
1 1 0 0 1 1 0 1 1 0 1 1 0 1
여러분이 만들 수 있는 1이 연속된 연속부분수열은

이며 그 길이는 8입니다.

입력

첫 번째 줄에 수열의 길이인 자연수 N(5<=N<100,000)이 주어집니다.
두 번째 줄에 N길이의 0과 1로 구성된 수열이 주어집니다.

출력

첫 줄에 최대 길이를 출력하세요.

예제 입력

14 2
1 1 0 0 1 1 0 1 1 0 1 1 0 1

예제 출력

8

문제 해결 방법

두 개의 포인터로 이 문제를 해결할 수 있다. count는 바꿀 수 있는 횟수를 나타내며, k로 초기화한다.
먼저, rt가 배열을 돌면서 0을 만나면 count를 하나 줄이고 1로 바꾼다. 바꿀 수 있는 최대 횟수 k만큼 모두 바꾸면 lt가 움직이면서 1로 바뀐 것을 다시 0으로 바꾼다. rt가 한 번 돌 때마다 answer와 (rt - lt + 1)을 비교하여 더 큰 값을 answer에 저장한다.

정답 풀이

public static int solution(int N, int k, int[] array) {
    int answer = 0;
    int lt = 0;
    int count = k;

    for (int rt = 0; rt < N; rt++) {
        if (array[rt] == 0) count--;
        while (count < 0) {
            if (array[lt] == 0) count++;
            lt++;
        }
        answer = Math.max(answer, rt - lt + 1);
    }

    return answer;
}