QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#601396 | #6339. Cookies | Dimash | 0 | 1ms | 5624kb | C++23 | 2.2kb | 2024-09-29 22:56:47 | 2024-09-29 22:56:47 |
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e2 + 12, MOD = (int)1e9 + 7;
int n, m, b[N], s = 0, val[N], c[N], p[N];
pair<int, int> a[N];
bool dp[N][N][N];
bool check(int j, int k) {
return (val[k] >= j);
}
void out(int res) {
vector<int> d;
int i = m, j = s, k = res;
while(j) {
if(dp[i - 1][j][k]) {
i--;
continue;
}
assert(j - b[i] >= 0);
j -= b[i];
k--;
d.push_back(b[i]);
if(dp[i - 1][j][k]) {
i--;
}
}
cout << res << '\n';
assert((int)d.size() == res);
for(int i:d) {
sort(a + 1, a + n + 1);
reverse(a + 1, a + n + 1);
cout << i << ' ';
for(int j = 1; j <= i; j++) {
a[j].first--;
cout << a[j].second << ' ';
}
cout << '\n';
}
}
void test() {
cin >> n;
for(int i = 1; i <= n; i++) {
cin >> a[i].first, c[i] = a[i].first;
s += c[i];
a[i].second = i;
}
sort(a + 1, a + n + 1);
sort(c + 1, c + n + 1);
for(int i = 1; i <= n; i++) {
p[i] = p[i - 1] + c[i];
}
int it = n;
for(int i = n; i >= 1; i--) {
while(it >= 1 && c[it] >= i) {
it--;
}
val[i] = p[it] + (n - it) * 1ll * i;
}
cin >> m;
for(int i = 1; i <= m; i++) {
cin >> b[i];
}
reverse(b + 1, b + m + 1);
dp[0][0][0] = 1;
for(int i = 1; i <= m; i++) {
for(int j = 1; j <= s; j++) {
for(int k = 1; k <= s; k++) {
dp[i][j][k] = dp[i - 1][j][k];
if(j - b[i] >= 0 && (dp[i - 1][j - b[i]][k - 1] || dp[i][j - b[i]][k - 1]) && check(j, k)) {
dp[i][j][k] = 1;
}
}
}
}
for(int i = 1; i <= s; i++) {
if(dp[m][s][i]) {
out(i);
return;
}
}
cout << -1 << '\n';
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
int t = 1;
// cin >> t;
while(t--)
test();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 6
Accepted
time: 0ms
memory: 3648kb
input:
1 1 1 1
output:
1 1 1
result:
ok good!
Test #2:
score: 6
Accepted
time: 0ms
memory: 3636kb
input:
2 1 1 1 1
output:
2 1 2 1 1
result:
ok good!
Test #3:
score: 6
Accepted
time: 0ms
memory: 3576kb
input:
2 1 1 1 2
output:
1 2 2 1
result:
ok good!
Test #4:
score: 6
Accepted
time: 0ms
memory: 3620kb
input:
2 1 1 2 1 2
output:
1 2 2 1
result:
ok good!
Test #5:
score: 0
Wrong Answer
time: 0ms
memory: 3884kb
input:
4 1 1 1 1 2 2 3
output:
-1
result:
wrong answer you didn't find a solution but jury did
Subtask #2:
score: 0
Wrong Answer
Test #28:
score: 0
Wrong Answer
time: 0ms
memory: 3652kb
input:
1 15 1 1
output:
-1
result:
wrong answer you didn't find a solution but jury did
Subtask #3:
score: 0
Wrong Answer
Test #45:
score: 0
Wrong Answer
time: 1ms
memory: 5624kb
input:
2 7 8 2 1 2
output:
-1
result:
wrong answer you didn't find a solution but jury did
Subtask #4:
score: 0
Skipped
Dependency #1:
0%
Subtask #5:
score: 0
Skipped
Dependency #4:
0%
Subtask #6:
score: 0
Skipped
Dependency #1:
0%