최대 길이 연속 부분수열

6. 최대 길이 연속부분수열

설명

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

📌two pointers

import java.util.*;
import java.util.Scanner;

class Main {
    public int solution(int n, int k, int[] arr) {
        //rt이동하면서 0인경우 발견시 k수만큼 cnt++, k보다 cnt가 커지면 
				//이제 lt 움직이면서 0인 곳 까지 간 다음에 cnt --;
        //1인 애들의 길이는 rt-lt+1, 예를 들어서 길이 1짜리는 lt rt 
				//모두 0번 인덱스에 있는 경우니
        int answer = 0, cnt = 0, lt = 0;
        for (int rt = 0; rt < n; rt++) {
            if (arr[rt] == 0) {
                cnt++;
            }
            while (cnt > k) {
                //lt가 이동한 칸이 0이면 cnt 깎고 이후 lt ++ 해주고 
								//answer에다가 길이 반환하는 느낌
                //그 다음엔 다시 위에 for문 가서 rt이동하다가 cnt 오버되면 
								//lt 이동하면서 반복!
                if (arr[lt] == 0) {
                    cnt--;
                }
                //if문 끝나고 넣기
                //0이면 cnt --하고 lt 다음칸, 아니면 그냥 다음 칸으로! 
                lt++;
            }
            answer = Math.max(answer, rt - lt + 1);
        }
        return answer;
    }

    public static void main(String[] args) {
        Main T = new Main();
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        System.out.print(T.solution(n, k, arr));
    }
}