[PS] 인프런 강의 - Two Pointers, Sliding window, Math 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이 연속된 연속부분수열은 1 1 0 0 1 1 1 1 1 1 1 1 0 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 void main(String[] args) {
		Scanner in = new Scanner(System.in);
		
		int n = in.nextInt();
		int k = in.nextInt();
		int[] arr = new int[n];
		
		for(int i=0; i<n; i++) {
			arr[i] = in.nextInt();
		}
		
		int answer = Integer.MIN_VALUE;
		int lt=0, cnt=0;
		for(int rt=0; rt<n; rt++) { //(1) rt 땡긴다
			if(arr[rt] == 0) cnt++; //(2) rt에 있는 값이 0이면 (2-1) cnt+1
			while(cnt > k) { //(2-2) cnt가 k보다 작아질 때까지 lt 땡긴다
				if(arr[lt] == 0) cnt--; //lt에 있는 값이 0이면 cnt-1
				lt++;
			}
			//(3) 길이값 최대인지 확인
			answer = Math.max(answer, rt-lt+1);
		}

		System.out.print(answer);
	}
}