#include <bits/stdc++.h>
//#define int long long
using namespace std;
const int N = 1.5e4 + 42;
vector<int> boite;
vector<bitset<N>> dp[N];
int n, m, a[N], sz[N], prefix[N];
void remonter(int i, int j, int sum) {
if(sum == 0)
return;
if(dp[i-1][j][sum - sz[j]]) {
boite.push_back(-sz[j]);
remonter(i-1, j, sum - sz[j]);
} else {
remonter(i, j+1, sum);
}
}
void printSol() {
vector<int> id[N];
for(int i = 0; i < n; i++)
for(int j = 0; j < a[i]; j++)
id[j].push_back(i+1);
queue<int> ids;
for(int i = 0; i < N; i++)
for(int j : id[i])
ids.push(j);
sort(boite.begin(), boite.end());
for(int b : boite) {
b = -b;
cout << b << ' ';
for(int i = 0; i < b; i++) {
cout << ids.front() << ' ';
ids.pop();
}
cout << '\n';
}
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> n;
for(int i = 0; i < n; i++) {
cin >> a[i];
for(int j = 0; j < N; j++)
prefix[j] += min(j, a[i]);
}
cin >> m;
dp[0].resize(m);
dp[0][m-1] = true;
for(int i = 0; i < m; i++)
cin >> sz[i];
int sum = prefix[n];
for(int i = 0; i <= sum; i++) {
for(int j = m-1; j >= 0; j--)
if(i * sz[j] <= sum) {
dp[i+1].resize(j+1);
j = 0;
}
for(int j = m-1; j >= 0; j--)
if(i * sz[j] <= sum) {
dp[i][j] <<= (N-prefix[i]-1);
dp[i][j] >>= (N-prefix[i]-1);
if(dp[i][j][sum]) {
remonter(i, j, sum);
cout << i << '\n';
printSol();
return 0;
}
if(j > 0)
dp[i][j-1] |= dp[i][j];
if((i+1) * sz[j] <= sum)
dp[i+1][j] |= (dp[i][j] << sz[j]);
}
}
cout << -1 << '\n';
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1108 KB |
Output is correct |
2 |
Correct |
1 ms |
1108 KB |
Output is correct |
3 |
Correct |
1 ms |
980 KB |
Output is correct |
4 |
Correct |
1 ms |
980 KB |
Output is correct |
5 |
Correct |
1 ms |
980 KB |
Output is correct |
6 |
Correct |
1 ms |
980 KB |
Output is correct |
7 |
Correct |
1 ms |
1108 KB |
Output is correct |
8 |
Correct |
7 ms |
2004 KB |
Output is correct |
9 |
Correct |
6 ms |
1108 KB |
Output is correct |
10 |
Correct |
8 ms |
4332 KB |
Output is correct |
11 |
Correct |
7 ms |
2020 KB |
Output is correct |
12 |
Correct |
7 ms |
2132 KB |
Output is correct |
13 |
Correct |
7 ms |
2280 KB |
Output is correct |
14 |
Correct |
7 ms |
2644 KB |
Output is correct |
15 |
Correct |
7 ms |
2592 KB |
Output is correct |
16 |
Correct |
7 ms |
2644 KB |
Output is correct |
17 |
Correct |
6 ms |
1252 KB |
Output is correct |
18 |
Correct |
6 ms |
1260 KB |
Output is correct |
19 |
Correct |
6 ms |
1128 KB |
Output is correct |
20 |
Correct |
6 ms |
1124 KB |
Output is correct |
21 |
Correct |
6 ms |
1644 KB |
Output is correct |
22 |
Correct |
8 ms |
4200 KB |
Output is correct |
23 |
Correct |
8 ms |
3816 KB |
Output is correct |
24 |
Correct |
6 ms |
1376 KB |
Output is correct |
25 |
Correct |
9 ms |
1380 KB |
Output is correct |
26 |
Correct |
6 ms |
1364 KB |
Output is correct |
27 |
Correct |
7 ms |
1256 KB |
Output is correct |
28 |
Correct |
7 ms |
2004 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
724 KB |
Output is correct |
2 |
Correct |
1 ms |
1108 KB |
Output is correct |
3 |
Correct |
1 ms |
980 KB |
Output is correct |
4 |
Correct |
1 ms |
980 KB |
Output is correct |
5 |
Correct |
8 ms |
2004 KB |
Output is correct |
6 |
Correct |
6 ms |
980 KB |
Output is correct |
7 |
Incorrect |
1 ms |
1108 KB |
Number of cookies does not match with A_i |
8 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1108 KB |
Output is correct |
2 |
Correct |
1 ms |
724 KB |
Output is correct |
3 |
Correct |
1 ms |
980 KB |
Output is correct |
4 |
Correct |
1 ms |
980 KB |
Output is correct |
5 |
Correct |
1 ms |
980 KB |
Output is correct |
6 |
Correct |
1 ms |
1108 KB |
Output is correct |
7 |
Correct |
1 ms |
980 KB |
Output is correct |
8 |
Correct |
1 ms |
1108 KB |
Output is correct |
9 |
Incorrect |
1 ms |
980 KB |
Number of cookies does not match with A_i |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1108 KB |
Output is correct |
2 |
Correct |
1 ms |
1108 KB |
Output is correct |
3 |
Correct |
1 ms |
980 KB |
Output is correct |
4 |
Correct |
1 ms |
980 KB |
Output is correct |
5 |
Correct |
1 ms |
980 KB |
Output is correct |
6 |
Correct |
1 ms |
980 KB |
Output is correct |
7 |
Correct |
1 ms |
1108 KB |
Output is correct |
8 |
Correct |
7 ms |
2004 KB |
Output is correct |
9 |
Correct |
6 ms |
1108 KB |
Output is correct |
10 |
Correct |
8 ms |
4332 KB |
Output is correct |
11 |
Correct |
7 ms |
2020 KB |
Output is correct |
12 |
Correct |
7 ms |
2132 KB |
Output is correct |
13 |
Correct |
7 ms |
2280 KB |
Output is correct |
14 |
Correct |
7 ms |
2644 KB |
Output is correct |
15 |
Correct |
7 ms |
2592 KB |
Output is correct |
16 |
Correct |
7 ms |
2644 KB |
Output is correct |
17 |
Correct |
6 ms |
1252 KB |
Output is correct |
18 |
Correct |
6 ms |
1260 KB |
Output is correct |
19 |
Correct |
6 ms |
1128 KB |
Output is correct |
20 |
Correct |
6 ms |
1124 KB |
Output is correct |
21 |
Correct |
6 ms |
1644 KB |
Output is correct |
22 |
Correct |
8 ms |
4200 KB |
Output is correct |
23 |
Correct |
8 ms |
3816 KB |
Output is correct |
24 |
Correct |
6 ms |
1376 KB |
Output is correct |
25 |
Correct |
9 ms |
1380 KB |
Output is correct |
26 |
Correct |
6 ms |
1364 KB |
Output is correct |
27 |
Correct |
7 ms |
1256 KB |
Output is correct |
28 |
Correct |
7 ms |
2004 KB |
Output is correct |
29 |
Correct |
1 ms |
1108 KB |
Output is correct |
30 |
Correct |
1 ms |
724 KB |
Output is correct |
31 |
Correct |
1 ms |
980 KB |
Output is correct |
32 |
Correct |
1 ms |
980 KB |
Output is correct |
33 |
Correct |
1 ms |
980 KB |
Output is correct |
34 |
Correct |
1 ms |
1108 KB |
Output is correct |
35 |
Correct |
1 ms |
980 KB |
Output is correct |
36 |
Correct |
1 ms |
1108 KB |
Output is correct |
37 |
Incorrect |
1 ms |
980 KB |
Number of cookies does not match with A_i |
38 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1108 KB |
Output is correct |
2 |
Correct |
1 ms |
1108 KB |
Output is correct |
3 |
Correct |
1 ms |
980 KB |
Output is correct |
4 |
Correct |
1 ms |
980 KB |
Output is correct |
5 |
Correct |
1 ms |
980 KB |
Output is correct |
6 |
Correct |
1 ms |
980 KB |
Output is correct |
7 |
Correct |
1 ms |
1108 KB |
Output is correct |
8 |
Correct |
7 ms |
2004 KB |
Output is correct |
9 |
Correct |
6 ms |
1108 KB |
Output is correct |
10 |
Correct |
8 ms |
4332 KB |
Output is correct |
11 |
Correct |
7 ms |
2020 KB |
Output is correct |
12 |
Correct |
7 ms |
2132 KB |
Output is correct |
13 |
Correct |
7 ms |
2280 KB |
Output is correct |
14 |
Correct |
7 ms |
2644 KB |
Output is correct |
15 |
Correct |
7 ms |
2592 KB |
Output is correct |
16 |
Correct |
7 ms |
2644 KB |
Output is correct |
17 |
Correct |
6 ms |
1252 KB |
Output is correct |
18 |
Correct |
6 ms |
1260 KB |
Output is correct |
19 |
Correct |
6 ms |
1128 KB |
Output is correct |
20 |
Correct |
6 ms |
1124 KB |
Output is correct |
21 |
Correct |
6 ms |
1644 KB |
Output is correct |
22 |
Correct |
8 ms |
4200 KB |
Output is correct |
23 |
Correct |
8 ms |
3816 KB |
Output is correct |
24 |
Correct |
6 ms |
1376 KB |
Output is correct |
25 |
Correct |
9 ms |
1380 KB |
Output is correct |
26 |
Correct |
6 ms |
1364 KB |
Output is correct |
27 |
Correct |
7 ms |
1256 KB |
Output is correct |
28 |
Correct |
7 ms |
2004 KB |
Output is correct |
29 |
Correct |
1 ms |
1108 KB |
Output is correct |
30 |
Correct |
1 ms |
724 KB |
Output is correct |
31 |
Correct |
1 ms |
980 KB |
Output is correct |
32 |
Correct |
1 ms |
980 KB |
Output is correct |
33 |
Correct |
1 ms |
980 KB |
Output is correct |
34 |
Correct |
1 ms |
1108 KB |
Output is correct |
35 |
Correct |
1 ms |
980 KB |
Output is correct |
36 |
Correct |
1 ms |
1108 KB |
Output is correct |
37 |
Incorrect |
1 ms |
980 KB |
Number of cookies does not match with A_i |
38 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1108 KB |
Output is correct |
2 |
Correct |
1 ms |
1108 KB |
Output is correct |
3 |
Correct |
1 ms |
980 KB |
Output is correct |
4 |
Correct |
1 ms |
980 KB |
Output is correct |
5 |
Correct |
1 ms |
980 KB |
Output is correct |
6 |
Correct |
1 ms |
980 KB |
Output is correct |
7 |
Correct |
1 ms |
1108 KB |
Output is correct |
8 |
Correct |
7 ms |
2004 KB |
Output is correct |
9 |
Correct |
6 ms |
1108 KB |
Output is correct |
10 |
Correct |
8 ms |
4332 KB |
Output is correct |
11 |
Correct |
7 ms |
2020 KB |
Output is correct |
12 |
Correct |
7 ms |
2132 KB |
Output is correct |
13 |
Correct |
7 ms |
2280 KB |
Output is correct |
14 |
Correct |
7 ms |
2644 KB |
Output is correct |
15 |
Correct |
7 ms |
2592 KB |
Output is correct |
16 |
Correct |
7 ms |
2644 KB |
Output is correct |
17 |
Correct |
6 ms |
1252 KB |
Output is correct |
18 |
Correct |
6 ms |
1260 KB |
Output is correct |
19 |
Correct |
6 ms |
1128 KB |
Output is correct |
20 |
Correct |
6 ms |
1124 KB |
Output is correct |
21 |
Correct |
6 ms |
1644 KB |
Output is correct |
22 |
Correct |
8 ms |
4200 KB |
Output is correct |
23 |
Correct |
8 ms |
3816 KB |
Output is correct |
24 |
Correct |
6 ms |
1376 KB |
Output is correct |
25 |
Correct |
9 ms |
1380 KB |
Output is correct |
26 |
Correct |
6 ms |
1364 KB |
Output is correct |
27 |
Correct |
7 ms |
1256 KB |
Output is correct |
28 |
Correct |
7 ms |
2004 KB |
Output is correct |
29 |
Correct |
1 ms |
724 KB |
Output is correct |
30 |
Correct |
1 ms |
1108 KB |
Output is correct |
31 |
Correct |
1 ms |
980 KB |
Output is correct |
32 |
Correct |
1 ms |
980 KB |
Output is correct |
33 |
Correct |
8 ms |
2004 KB |
Output is correct |
34 |
Correct |
6 ms |
980 KB |
Output is correct |
35 |
Incorrect |
1 ms |
1108 KB |
Number of cookies does not match with A_i |
36 |
Halted |
0 ms |
0 KB |
- |