Submission #1294855

#TimeUsernameProblemLanguageResultExecution timeMemory
1294855nguyenkhangninh99Shopping Plans (CCO20_day2problem3)C++20
25 / 25
271 ms76540 KiB
#include <bits/stdc++.h> using namespace std; #define int long long const int inf = 1E18; struct modify_online_subset_minimum{ vector<int> a, ans; int l, r; priority_queue<array<int, 4>, vector<array<int, 4>>, greater<array<int, 4>>> pq; modify_online_subset_minimum(vector<int> a_, int l_, int r_) : l(l_), r(r_){ a = a_; sort(a.begin(), a.end()); if(l == 0) pq.push({0, -1, -1, -1}); int val = 0; for(int i = 0; i < a.size(); i++){ val += a[i]; if(i + 1 >= l && i + 1 <= r) pq.push({val, i, i, a.size()}); } } int operator[](int i){ while(i >= ans.size()){ if(pq.empty()) break; auto [v, c, i, h] = pq.top(); pq.pop(); ans.push_back(v); if(c == -1) continue; if(i + 1 < h) pq.push({v - a[i] + a[i + 1], c, i + 1, h}); if(c > 0 && c < i) pq.push({v - a[c - 1] + a[c], c - 1, c, i}); } return i < ans.size() ? ans[i] : inf; } }; void solve() { int n, m, k; cin >> n >> m >> k; vector<vector<int>> a(m); for(int i = 0; i < n; i++){ int u, t; cin >> t >> u; a[t - 1].push_back(u); } vector<modify_online_subset_minimum> all; for(int i = 0; i < m; i++){ int l, r; cin >> l >> r; all.push_back(modify_online_subset_minimum(a[i], l, r)); } sort(all.begin(), all.end(), [&](auto& u, auto& v){ return u[1] - u[0] < v[1] - v[0]; }); priority_queue<array<int, 3>, vector<array<int, 3>>, greater<array<int, 3>>> pq; int val = 0; for(int i = 0; i < m; i++){ val += all[i][0]; if(all[i][0] == inf){ while(k--) cout << -1 << '\n'; return; } } pq.push({val, -1, 0}); while(k--){ if(pq.empty() || pq.top()[0] >= inf){ cout << "-1\n"; continue; } auto [v, i, p] = pq.top(); pq.pop(); cout << v << '\n'; if(i != -1) pq.push({v + all[i][p + 1] - all[i][p], i, p + 1}); if(i + 1 < m) pq.push({v + all[i + 1][1] - all[i + 1][0], i + 1, 1}); if(p == 1 && i + 1 < m) pq.push({v + all[i + 1][1] - all[i + 1][0] + all[i][0] - all[i][1], i + 1, 1}); } } signed main(){ ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); solve(); }

Compilation message (stderr)

Main.cpp: In constructor 'modify_online_subset_minimum::modify_online_subset_minimum(std::vector<long long int>, long long int, long long int)':
Main.cpp:20:68: warning: narrowing conversion of '((modify_online_subset_minimum*)this)->modify_online_subset_minimum::a.std::vector<long long int>::size()' from 'std::vector<long long int>::size_type' {aka 'long unsigned int'} to 'long long int' [-Wnarrowing]
   20 |             if(i + 1 >= l && i + 1 <= r) pq.push({val, i, i, a.size()});
      |                                                              ~~~~~~^~
#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...