#include <bits/stdc++.h>
#define f0(i, n) for(ll i(0); i < (n); i++)
#define f1(i, n) for(ll i(1); i <= n; i++)
using namespace std;
typedef long long ll;
const ll N = 300002;
ll n, m, s, a[N], b[N], id[N], ans1[N], ans2[N];
int main(){
ios_base::sync_with_stdio(0);
cin >> n >> m >> s;
f1(i, n){
cin >> a[i] >> b[i];
id[i] = i;
}
sort(id + 1, id + n + 1, [&](ll a1, ll a2){
return a[a1] - b[a1] > a[a2] - b[a2];
});
priority_queue <ll, vector <ll>, greater <ll> > q;
f1(i, n){
if(q.size() < m){
ans1[i] = ans1[i - 1] + a[id[i]];
q.push(a[id[i]]);
}
else{
ll x = q.top();
q.pop();
q.push(a[id[i]]);
ans1[i] = ans1[i - 1] - x + a[id[i]];
}
}
while(q.size()) q.pop();
for(ll i = n; i >= 1; i--){
if(q.size() < s){
ans2[i] = ans2[i + 1] + b[id[i]];
q.push(b[id[i]]);
}
else{
ll x = q.top();
q.pop();
q.push(b[id[i]]);
ans2[i] = ans2[i + 1] - x + b[id[i]];
}
}
f1(i, n) ans1[i] = max(ans1[i], ans1[i - 1]);
for(ll i = n; i >= 1; i--) ans2[i] = max(ans2[i + 1], ans2[i]);
ll res = LLONG_MIN;
f1(i, n){
if(i >= m && n - i >= s) res = max(res, ans1[i] + ans2[i +1]);
}
cout << res;
}
Compilation message
school.cpp: In function 'int main()':
school.cpp:24:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if(q.size() < m){
~~~~~~~~~^~~
school.cpp:37:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if(q.size() < s){
~~~~~~~~~^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
Output is correct |
2 |
Runtime error |
3 ms |
892 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
3 |
Correct |
2 ms |
892 KB |
Output is correct |
4 |
Correct |
2 ms |
892 KB |
Output is correct |
5 |
Correct |
2 ms |
892 KB |
Output is correct |
6 |
Correct |
2 ms |
892 KB |
Output is correct |
7 |
Correct |
4 ms |
976 KB |
Output is correct |
8 |
Correct |
5 ms |
1012 KB |
Output is correct |
9 |
Correct |
5 ms |
1020 KB |
Output is correct |
10 |
Correct |
4 ms |
1036 KB |
Output is correct |
11 |
Correct |
5 ms |
1036 KB |
Output is correct |
12 |
Correct |
4 ms |
1036 KB |
Output is correct |
13 |
Correct |
23 ms |
2572 KB |
Output is correct |
14 |
Correct |
56 ms |
4152 KB |
Output is correct |
15 |
Correct |
82 ms |
7096 KB |
Output is correct |
16 |
Correct |
99 ms |
9648 KB |
Output is correct |
17 |
Correct |
136 ms |
10268 KB |
Output is correct |
18 |
Correct |
155 ms |
11052 KB |
Output is correct |
19 |
Correct |
166 ms |
11952 KB |
Output is correct |
20 |
Correct |
174 ms |
13496 KB |
Output is correct |