# |
제출 시각 |
아이디 |
문제 |
언어 |
결과 |
실행 시간 |
메모리 |
789302 |
2023-07-21T09:04:44 Z |
박영우(#10045) |
Cookies (JOI23_cookies) |
C++17 |
|
50 ms |
1748 KB |
#include <bits/stdc++.h>
#include <cassert>
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx,avx2,fma")
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
#define MAX 600
#define MAXS 20
#define INF 1000000000000000001
#define bb ' '
#define ln '\n'
#define Ln '\n'
int A[MAX];
int cnt[MAX];
int B[MAX];
int dp[MAX][MAX];
vector<int> ansv;
int N, M;
void print() {
vector<int> v;
vector<vector<int>> res;
int i;
for (i = 1; i <= N; i++) v.push_back(i);
for (auto b : ansv) {
vector<int> nv;
sort(v.begin(), v.end(), [&](int i, int j) {return A[i] > A[j]; });
for (i = 0; i < b; i++) {
nv.push_back(v[i]);
A[v[i]]--;
}
res.push_back(nv);
}
for (i = 1; i <= N; i++) if (A[i]) {
cout << -1 << ln;
exit(0);
}
cout << res.size() << ln;
for (auto& v : res) {
cout << v.size() << bb;
for (auto x : v) cout << x << bb;
cout << ln;
}
}
signed main() {
ios::sync_with_stdio(false), cin.tie(0);
cin >> N;
int i, j, k;
int S = 0;
for (i = 1; i <= N; i++) {
cin >> A[i];
S += A[i];
for (j = 1; j <= A[i]; j++) cnt[j]++;
}
for (i = 1; i <= S; i++) cnt[i] += cnt[i - 1];
cin >> M;
for (i = 1; i <= M; i++) cin >> B[i];
if (M == 1) {
if (S % B[1]) {
cout << 0 << ln;
return 0;
}
ansv = vector<int>(S / B[1], B[1]);
print();
return 0;
}
memset(dp, -1, sizeof(dp));
dp[0][0] = 1e9;
for (i = 1; i <= S; i++) {
for (j = 0; j <= cnt[i - 1]; j++) {
if (!~dp[i - 1][j]) continue;
for (k = 1; k <= M; k++) if (j + B[k] <= cnt[i] && B[k] <= dp[i - 1][j]) dp[i][j + B[k]] = max(dp[i][j + B[k]], B[k]);
}
}
int loc;
for (i = 1; i <= S; i++) if (~dp[i][S]) break;
loc = i;
if (loc > S) {
cout << -1;
return 0;
}
int n = S;
for (i = loc; i >= 1; i--) {
int path = dp[i][n];
assert(path != -1);
n -= path;
if (path) ansv.push_back(path);
}
sort(ansv.begin(), ansv.end(), [&](int i, int j) {return i > j; });
print();
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1620 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
1620 KB |
Output is correct |
6 |
Correct |
1 ms |
1620 KB |
Output is correct |
7 |
Correct |
1 ms |
1620 KB |
Output is correct |
8 |
Correct |
4 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
50 ms |
1720 KB |
Output is correct |
11 |
Correct |
1 ms |
1620 KB |
Output is correct |
12 |
Correct |
4 ms |
1620 KB |
Output is correct |
13 |
Correct |
5 ms |
1736 KB |
Output is correct |
14 |
Correct |
2 ms |
1620 KB |
Output is correct |
15 |
Correct |
2 ms |
1748 KB |
Output is correct |
16 |
Correct |
1 ms |
1620 KB |
Output is correct |
17 |
Correct |
1 ms |
1620 KB |
Output is correct |
18 |
Correct |
2 ms |
1748 KB |
Output is correct |
19 |
Correct |
1 ms |
1620 KB |
Output is correct |
20 |
Correct |
1 ms |
1748 KB |
Output is correct |
21 |
Correct |
2 ms |
1620 KB |
Output is correct |
22 |
Correct |
7 ms |
1620 KB |
Output is correct |
23 |
Correct |
4 ms |
1620 KB |
Output is correct |
24 |
Correct |
2 ms |
1620 KB |
Output is correct |
25 |
Correct |
2 ms |
1620 KB |
Output is correct |
26 |
Correct |
1 ms |
1668 KB |
Output is correct |
27 |
Correct |
1 ms |
1620 KB |
Output is correct |
28 |
Correct |
2 ms |
1620 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
Number of cookies does not match with A_i |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1620 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
Number of cookies does not match with A_i |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1620 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
1620 KB |
Output is correct |
6 |
Correct |
1 ms |
1620 KB |
Output is correct |
7 |
Correct |
1 ms |
1620 KB |
Output is correct |
8 |
Correct |
4 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
50 ms |
1720 KB |
Output is correct |
11 |
Correct |
1 ms |
1620 KB |
Output is correct |
12 |
Correct |
4 ms |
1620 KB |
Output is correct |
13 |
Correct |
5 ms |
1736 KB |
Output is correct |
14 |
Correct |
2 ms |
1620 KB |
Output is correct |
15 |
Correct |
2 ms |
1748 KB |
Output is correct |
16 |
Correct |
1 ms |
1620 KB |
Output is correct |
17 |
Correct |
1 ms |
1620 KB |
Output is correct |
18 |
Correct |
2 ms |
1748 KB |
Output is correct |
19 |
Correct |
1 ms |
1620 KB |
Output is correct |
20 |
Correct |
1 ms |
1748 KB |
Output is correct |
21 |
Correct |
2 ms |
1620 KB |
Output is correct |
22 |
Correct |
7 ms |
1620 KB |
Output is correct |
23 |
Correct |
4 ms |
1620 KB |
Output is correct |
24 |
Correct |
2 ms |
1620 KB |
Output is correct |
25 |
Correct |
2 ms |
1620 KB |
Output is correct |
26 |
Correct |
1 ms |
1668 KB |
Output is correct |
27 |
Correct |
1 ms |
1620 KB |
Output is correct |
28 |
Correct |
2 ms |
1620 KB |
Output is correct |
29 |
Correct |
1 ms |
1620 KB |
Output is correct |
30 |
Incorrect |
0 ms |
212 KB |
Number of cookies does not match with A_i |
31 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1620 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
1620 KB |
Output is correct |
6 |
Correct |
1 ms |
1620 KB |
Output is correct |
7 |
Correct |
1 ms |
1620 KB |
Output is correct |
8 |
Correct |
4 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
50 ms |
1720 KB |
Output is correct |
11 |
Correct |
1 ms |
1620 KB |
Output is correct |
12 |
Correct |
4 ms |
1620 KB |
Output is correct |
13 |
Correct |
5 ms |
1736 KB |
Output is correct |
14 |
Correct |
2 ms |
1620 KB |
Output is correct |
15 |
Correct |
2 ms |
1748 KB |
Output is correct |
16 |
Correct |
1 ms |
1620 KB |
Output is correct |
17 |
Correct |
1 ms |
1620 KB |
Output is correct |
18 |
Correct |
2 ms |
1748 KB |
Output is correct |
19 |
Correct |
1 ms |
1620 KB |
Output is correct |
20 |
Correct |
1 ms |
1748 KB |
Output is correct |
21 |
Correct |
2 ms |
1620 KB |
Output is correct |
22 |
Correct |
7 ms |
1620 KB |
Output is correct |
23 |
Correct |
4 ms |
1620 KB |
Output is correct |
24 |
Correct |
2 ms |
1620 KB |
Output is correct |
25 |
Correct |
2 ms |
1620 KB |
Output is correct |
26 |
Correct |
1 ms |
1668 KB |
Output is correct |
27 |
Correct |
1 ms |
1620 KB |
Output is correct |
28 |
Correct |
2 ms |
1620 KB |
Output is correct |
29 |
Correct |
1 ms |
1620 KB |
Output is correct |
30 |
Incorrect |
0 ms |
212 KB |
Number of cookies does not match with A_i |
31 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1620 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
1620 KB |
Output is correct |
6 |
Correct |
1 ms |
1620 KB |
Output is correct |
7 |
Correct |
1 ms |
1620 KB |
Output is correct |
8 |
Correct |
4 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
50 ms |
1720 KB |
Output is correct |
11 |
Correct |
1 ms |
1620 KB |
Output is correct |
12 |
Correct |
4 ms |
1620 KB |
Output is correct |
13 |
Correct |
5 ms |
1736 KB |
Output is correct |
14 |
Correct |
2 ms |
1620 KB |
Output is correct |
15 |
Correct |
2 ms |
1748 KB |
Output is correct |
16 |
Correct |
1 ms |
1620 KB |
Output is correct |
17 |
Correct |
1 ms |
1620 KB |
Output is correct |
18 |
Correct |
2 ms |
1748 KB |
Output is correct |
19 |
Correct |
1 ms |
1620 KB |
Output is correct |
20 |
Correct |
1 ms |
1748 KB |
Output is correct |
21 |
Correct |
2 ms |
1620 KB |
Output is correct |
22 |
Correct |
7 ms |
1620 KB |
Output is correct |
23 |
Correct |
4 ms |
1620 KB |
Output is correct |
24 |
Correct |
2 ms |
1620 KB |
Output is correct |
25 |
Correct |
2 ms |
1620 KB |
Output is correct |
26 |
Correct |
1 ms |
1668 KB |
Output is correct |
27 |
Correct |
1 ms |
1620 KB |
Output is correct |
28 |
Correct |
2 ms |
1620 KB |
Output is correct |
29 |
Incorrect |
1 ms |
212 KB |
Number of cookies does not match with A_i |
30 |
Halted |
0 ms |
0 KB |
- |