최대 길이 연속부분수열

설명

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로 구성된 수열이 주어집니다.

 

출력

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

예시 입력 1 

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

 

예시 출력 1

8

 

소스 코드 

import java.util.Scanner;

public class Main {
    public static int solution(int N, int k, int[] arr) {
        int ans = 0;
        int cnt = 0, lt = 0; //cnt는 0의 개수
        for (int rt = 0; rt < N; rt++) {
            if (arr[rt] == 0) { //만약 arr의 인덱스rt의 값이 0이라면 cnt 카운트 
                cnt++;
            }
            //0이 최대 k개 사용 가능하므로 k개를 초과하면 0 하나를 뺴줘야한다.
            while (cnt > k) {
                //이때 lt를 한칸씩 올리며 0을 찾고, 0을 찾았을 시 cnt를 감소시킨다.
                if (arr[lt] == 0) {
                    cnt--;
                }
                lt++;
            }
            ans = Math.max(ans, rt - lt + 1); // 기존 길이의 최댓값과 현재 길이의 최댓값과 비교하여 더 큰 값 대입
        }
        return ans;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int k = sc.nextInt();
        int[] arr = new int[N];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = sc.nextInt();
        }

        System.out.println(solution(N, k, arr));
    }
}