[Inflearn] 자바 알고리즘 문제풀이 #03-06 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로 구성된 수열이 주어집니다.

 

출력 

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

 

 

 

선생님 풀이

import java.util.Scanner;

public class Main3_6 {
    public int Solution(int n, int k, int[] arr) {
        int answer=0, cnt=0, lt=0;
        for (int rt = 0; rt < n; rt++) {
            if(arr[rt]==0) cnt++;
            while(cnt>k) {
                if(arr[lt]==0) cnt--;
                lt++;
            }
            answer = Math.max(answer, rt-lt+1);
        }
        return answer;
    }

    public static void main(String[] args) {
        Main3_6 T = new Main3_6();
        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.println(T.Solution(n, k, arr));
    }
}

sliding window, two pointer는 익숙해졌다고 생각했는데 이렇게 간단한 걸 왜 못풀었을까........

요즘 문제 잘 풀어서 기분 좋았는데 한 순간에 나락감(?)

점심먹고 3장 문제들 복습하면서 sliding window, two pointer와 친해지는 시간을 가져야 겠다... 딱대 요놈들아.

 

 

 

결과