File tree Expand file tree Collapse file tree 1 file changed +40
-0
lines changed
Expand file tree Collapse file tree 1 file changed +40
-0
lines changed Original file line number Diff line number Diff line change 1+ ```java
2+ import java.io.*;
3+ import java.util.*;
4+
5+ public class Main {
6+ public static void main(String[] args) throws IOException {
7+ BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
8+
9+ int T = Integer.parseInt(st.nextToken());
10+ int W = Integer.parseInt(st.nextToken());
11+
12+ int[] tree = new int[T + 1];
13+ for (int i = 1; i <= T; i++) {
14+ tree[i] = Integer.parseInt(br.readLine().trim());
15+ }
16+
17+ int[][] dp = new int[T + 1][W + 1];
18+
19+ for (int t = 1; t <= T; t++) {
20+ for (int w = 0; w <= W; w++) {
21+ int pos = (w % 2 == 0) ? 1 : 2;
22+ int get = (tree[t] == pos) ? 1 : 0;
23+
24+ if (w == 0) {
25+ dp[t][w] = dp[t - 1][w] + get;
26+ } else {
27+ dp[t][w] = Math.max(dp[t - 1][w], dp[t - 1][w - 1]) + get;
28+ }
29+ }
30+ }
31+
32+ int ans = 0;
33+ for (int w = 0; w <= W; w++) {
34+ ans = Math.max(ans, dp[T][w]);
35+ }
36+
37+ System.out.println(ans);
38+ }
39+ }
40+ ```
You can’t perform that action at this time.
0 commit comments