#include <bits/stdc++.h>
#define ll long long
#define ull unsigned ll
#define f first
#define s second
#define pii pair<ll,ll>
#define pll pair<ll,ll>
#define pb push_back
#define epb emplace_back
using namespace std;
const int nmax =5e5 + 1;
const ll linf = 1e18 + 1;
ll DP[nmax], dp[nmax];
ll A[nmax], B[nmax];
int main(){
// freopen("school.in", "r", stdin);
// freopen("school.out", "w", stdout);
int n; cin >> n;
int m; cin >> m;
int s; cin >> s;
for(int i = 1; i <= n; i++)
cin >> A[i] >> B[i];
vector <pll> vec;
for(int i = 1; i <= n; i++){
vec.pb({B[i] - A[i], i});
}
sort(vec.begin(), vec.end());
multiset <ll> cur;
for(int i = 0; i < vec.size(); i++){
int x = vec[i].s;
if(cur.size() < m)
DP[i + 1] = DP[i] + A[x], cur.insert(A[x]);
else{
DP[i + 1] = DP[i];
if(A[x] < *cur.begin()) continue;
DP[i + 1] = DP[i] - *cur.begin() + A[x];
cur.insert(A[x]);
cur.erase(cur.begin());
}
}
cur.clear();
for(int i = vec.size() - 1; i >= 0; i--){
int x = vec[i].s;
if(cur.size() < s)
dp[i + 1] = dp[i + 2] + B[x], cur.insert(B[x]);
else{
dp[i + 1] = dp[i + 2];
if(B[x] < *cur.begin()) continue;
dp[i + 1] = dp[i + 2] - *cur.begin() + B[x];
cur.insert(B[x]);
cur.erase(cur.begin());
}
}
ll ans = 0;
for(int i = 0; i <= n; i++)
ans = max(ans, DP[i] + dp[i + 1]);
cout << ans;
return 0;
}
Compilation message
school.cpp: In function 'int main()':
school.cpp:31:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
31 | for(int i = 0; i < vec.size(); i++){
| ~~^~~~~~~~~~~~
school.cpp:33:23: warning: comparison of integer expressions of different signedness: 'std::multiset<long long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
33 | if(cur.size() < m)
| ~~~~~~~~~~~^~~
school.cpp:46:23: warning: comparison of integer expressions of different signedness: 'std::multiset<long long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
46 | if(cur.size() < s)
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Incorrect |
1 ms |
340 KB |
Output isn't correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
1 ms |
340 KB |
Output is correct |
7 |
Correct |
4 ms |
724 KB |
Output is correct |
8 |
Correct |
4 ms |
724 KB |
Output is correct |
9 |
Correct |
4 ms |
724 KB |
Output is correct |
10 |
Correct |
4 ms |
724 KB |
Output is correct |
11 |
Correct |
5 ms |
768 KB |
Output is correct |
12 |
Correct |
6 ms |
724 KB |
Output is correct |
13 |
Correct |
32 ms |
3776 KB |
Output is correct |
14 |
Correct |
71 ms |
5828 KB |
Output is correct |
15 |
Correct |
103 ms |
9852 KB |
Output is correct |
16 |
Correct |
189 ms |
18096 KB |
Output is correct |
17 |
Correct |
240 ms |
18076 KB |
Output is correct |
18 |
Correct |
254 ms |
18744 KB |
Output is correct |
19 |
Correct |
271 ms |
20288 KB |
Output is correct |
20 |
Correct |
299 ms |
23216 KB |
Output is correct |