문제
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로 구성된 수열이 주어집니다.
출력
첫 줄에 최대 길이를 출력하세요.
예시 입력
14 2
1 1 0 0 1 1 0 1 1 0 1 1 0 1
예시 출력
8
내 풀이
private static int getLongest(int num, int k, int[] nums) {
// 0의 위치의 인덱스들을 갖는 List
List<Integer> indicesOfZero = new ArrayList<>();
indicesOfZero.add(-1); // 맨 처음 앞은 0으로 계산
for (int i = 0; i < num; i++) {
if (nums[i] == 0) indicesOfZero.add(i);
}
indicesOfZero.add(num); // 맨 마지막 뒤는 0으로 계산
// System.out.println(indicesOfZero);
int max = 0;
for (int i = 0; i < indicesOfZero.size() - (k + 1); i++) {
// 0...0...0...0
// 첫 0부터 (그 다음 (k+1)번쨰 0의 위치-1)이 최대연속길이
int firstZeroIndex = indicesOfZero.get(i);
int forthZeroIndex = indicesOfZero.get(i + (k + 1));
int length = (forthZeroIndex -1) - firstZeroIndex;
max = Math.max(max, length);
}
return max;
}
다른 풀이
/* 강의 풀이 */
public static int solution(int n, int k, int[] arr) {
int max = 0, numOfZero = 0, left = 0;
for (int right = 0; right < n; right++) {
// 오른쪽에서 0을 만났는데 k개 초과면 왼쪽에서 0을 뺀다
if (arr[right] == 0) numOfZero++;
while (numOfZero > k) {
if (arr[left] == 0) numOfZero--;
left++;
}
// 최대길이는 매번 확인한다 (맨 처음 앞과 맨 마지막 뒤 신경안써도 됨)
max = Math.max(max, right - left + 1);
}
return max;
}
'Java > Coding-Test' 카테고리의 다른 글
[Java] 4-3 매출액의 종류 (0) | 2022.01.23 |
---|---|
[Java] 4-1 학급 회장(해쉬) (0) | 2022.01.22 |
[Java] 3-5 연속된 자연수의 합 (0) | 2022.01.22 |
[Java] 3-4 연속 부분수열 (0) | 2022.01.20 |
[Java] 3-3 최대 매출 (0) | 2022.01.20 |
댓글