https://www.acmicpc.net/problem/22857
사용 알고리즘
- 다이나믹 프로그래밍
풀이
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
/*
테이블: d[i][j]: 첫 i개의 원소 중에서 j개의 홀수를 삭제했을 때 연속되는 짝수길이
점화식:
* nums[i-1]이 짝수: d[
*/
public class Main {
static int n, k;
static int [] nums;
static int [][] d;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
k = Integer.parseInt(st.nextToken());
nums = new int[n];
d = new int[n + 1][k + 1];
st = new StringTokenizer(br.readLine());
for(int i = 0; i < n; i++) {
nums[i] = Integer.parseInt(st.nextToken());
}
int max = 0;
for(int i = 1; i <= n; i++) { // i-1이니까 n을 포함해야함
for(int j = 0; j <= k; j++) {
if (nums[i-1] % 2 == 0) { // 짝수일 때
d[i][j] = d[i-1][j] + 1;
}
else { // 홀일 때
if (j > 0)
d[i][j] = d[i-1][j-1];
}
max = Math.max(max, d[i][j]);
}
}
System.out.println(max);
}
}
회고
테이블
d[i][j]: 첫 i개 원소중에서 j개의 홀수를 제거했을 때 연속되는 짝수 수열의 길이
점화식
nums[i-1]이 홀수일 때: d[i] = d[i-1][j] + 1
nums[i-1]이 짝수일 때: d[i] = d[i-1][j-1]
2차원 배열로 DP를 풀이를 시도했는데, 풀리지가 않아 답을 참고했다.
nums[i-1]을 보고 홀수 짝수 판별을 하는 아이디어가 중요하다