Submission #1094425

#TimeUsernameProblemLanguageResultExecution timeMemory
1094425_8_8_Cookies (JOI23_cookies)C++17
63 / 100
5 ms2140 KiB
#include <bits/stdc++.h> using namespace std; typedef long long ll; const int N = 15e2 + 12, MOD = (int)1e9 + 7; int n, m, b[N], s = 0, val[N], c[N], p[N]; pair<int, int> a[N]; vector<bitset<N>> dp[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(k < (int)dp[i - 1].size() && dp[i - 1][k][j]) { i--; continue; } assert(j - b[i] >= 0); j -= b[i]; k--; d.push_back(b[i]); if(k < (int)dp[i - 1].size() && dp[i - 1][k][j]) { 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 = s; 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); for(int i = 0; i <= m; ++i) { int t = (i ? (s + b[i] - 1) / b[i] + 1 : 1) + 10; dp[i].resize(t); dp[i][0][0] = 1; } for(int i = 1; i <= m; i++) { for(int k = 1; k < dp[i].size(); k++) { for(int j = k * b[i]; j <= s; j++) { if((int)dp[i - 1].size() > k) dp[i][k][j] = dp[i - 1][k][j]; if(j - b[i] >= 0 && ((dp[i - 1].size() > k - 1 && dp[i - 1][k - 1][j - b[i]]) || dp[i][k - 1][j - b[i]]) && check(j, k)) { dp[i][k][j] = 1; } } } } for(int i = 1; i < dp[m].size(); i++) { if(dp[m][i][s]) { 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; }

Compilation message (stderr)

cookies.cpp: In function 'void test()':
cookies.cpp:73:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::bitset<1512> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   73 |         for(int k = 1; k < dp[i].size(); k++) {
      |                        ~~^~~~~~~~~~~~~~
cookies.cpp:76:56: warning: comparison of integer expressions of different signedness: 'std::vector<std::bitset<1512> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   76 |                 if(j - b[i] >= 0 && ((dp[i - 1].size() > k - 1 && dp[i - 1][k - 1][j - b[i]]) || dp[i][k - 1][j - b[i]]) && check(j, k)) {
      |                                       ~~~~~~~~~~~~~~~~~^~~~~~~
cookies.cpp:82:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::bitset<1512> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |     for(int i = 1; i < dp[m].size(); i++) {
      |                    ~~^~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...