코딩 테스트/3. Tow pointers, Sliding window

Q3 - 6 최대 길이 연속부분수열

길동이이이잉 2021. 9. 8. 21:20
728x90
반응형

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

 

 

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		int in1 = in.nextInt();
		int in2 = in.nextInt();
		int[] in3 = new int[in1];
		for(int i = 0; i < in1; i++) {
			in3[i] = in.nextInt();
		}
		
		int lt = 0, answer = 0, cnt = 0, leg = 0;
		for(int rt = 0; rt < in1; rt++) {
			if(in3[rt] == 0) {
				cnt++;
				while(cnt > in2) {
					if(in3[lt] == 0) cnt--;
					lt++;
				}
			}
			leg = rt - lt + 1;
			answer = Math.max(leg, answer);
		}
		System.out.println(answer);
		in.close();
		return;
	}

}
728x90
반응형