-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathBOJ_1006.cpp
More file actions
78 lines (70 loc) · 1.72 KB
/
BOJ_1006.cpp
File metadata and controls
78 lines (70 loc) · 1.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
int n, w;
vector<vector<int>> arr;
vector<vector<int>> dp;
int search_dp(int r, int c) {
if (c < 0)
return 0;
if (dp[r][c] != -1)
return dp[r][c];
if (r == 2) {
dp[r][c] = min(search_dp(0, c) + 1, search_dp(1, c) + 1);
if (arr[0][c] + arr[1][c] <= w) {
dp[r][c] = min(dp[r][c], search_dp(2, c - 1) + 1);
}
if (c > 0 && arr[0][c - 1] + arr[0][c] <= w &&
arr[1][c - 1] + arr[1][c] <= w) {
dp[r][c] = min(dp[r][c], search_dp(2, c - 2) + 2);
}
} else {
dp[r][c] = search_dp(2, c - 1) + 1;
if (c > 0 && arr[r][c - 1] + arr[r][c] <= w) {
dp[r][c] = min(dp[r][c], search_dp(1 - r, c - 1) + 1);
}
}
return dp[r][c];
}
int main(void) {
int tc;
cin >> tc;
while (tc--) {
cin >> n >> w;
arr.assign(2, vector<int>(n, 0));
for (int i = 0; i < 2; i++) {
for (int j = 0; j < n; j++) {
cin >> arr[i][j];
}
}
dp.assign(3, vector<int>(n, -1));
int a = search_dp(2, n - 1);
if (n == 1) {
cout << a << '\n';
continue;
}
int b = 987654321;
if (arr[0][0] + arr[0][n - 1] <= w) {
dp.assign(3, vector<int>(n, -1));
dp[0][0] = 0;
b = search_dp(1, n - 1) + 1;
}
int c = 987654321;
if (arr[1][0] + arr[1][n - 1] <= w) {
dp.assign(3, vector<int>(n, -1));
dp[1][0] = 0;
c = search_dp(0, n - 1) + 1;
}
int d = 987654321;
if (arr[1][0] + arr[1][n - 1] <= w && arr[0][0] + arr[0][n - 1] <= w) {
dp.assign(3, vector<int>(n, -1));
dp[0][0] = 0;
dp[1][0] = 0;
dp[2][0] = 0;
d = search_dp(2, n - 2) + 2;
}
cout << min({a, b, c, d}) << "\n";
}
return 0;
}