728x90
반응형
▷ 문제
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로 구성된 수열이 주어집니다.
* 출력
첫 줄에 최대 길이를 출력하세요.
▷ 입력 예시
14 2
1 1 0 0 1 1 0 1 1 0 1 1 0 1
▷ 출력 예시
8
▷ 풀이
import java.util.Scanner;
public class Main {
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){
Main main = new Main();
Scanner kb = new Scanner(System.in);
int n = kb.nextInt();
int k = kb.nextInt();
int[] arr = new int[n];
for(int i=0; i<n; i++){
arr[i] = kb.nextInt();
}
System.out.println(main.solution(n, k, arr));
}
}
반응형
'Algorithm > Two pointers, Sliding window' 카테고리의 다른 글
[알고리즘]입력받은 수가 연속된 자연수의 합으로 이루어지는 경우의 수 (0) | 2021.10.18 |
---|---|
[알고리즘]연속 부분수열 (0) | 2021.10.18 |
[알고리즘]최대 매출 - 배열에서 연속된 요소 N개의 최대 합 구하기 (0) | 2021.10.18 |
[알고리즘]두 배열의 공통 원소 출력하기 (0) | 2021.10.18 |
[알고리즘]두 배열 합치기 및 오름차순 정렬 (0) | 2021.10.17 |