Submission #635109

#	Time	Username	Problem	Language	Result	Execution time	Memory
635109		pragmatist	Schools (IZhO13_school)	C++17	35 / 100	2085 ms	29524 KiB

school

#include<bits/stdc++.h>
 
#define ll long long
#define pb push_back
#define x first
#define y second
#define sz(v) (int)v.size()
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define ld long double              
#define nl "\n"

using namespace std;
using pii = pair<int, int>;
 
const int N = (int)3e5 + 7;
const int inf = (int)1e9 + 7;
const ll INF = (ll)1e18 + 7;
const int MOD = (int)1e9 + 7;
const int M = (int)5e4 + 7; 

int n, m, s, a[N], b[N];
bool used[N];

void solve() {
	cin >> n >> m >> s;
	vector<pair<pii, int> > v;
	for(int i = 1; i <= n; ++i) {
		cin >> a[i] >> b[i];
		v.pb({{a[i], -b[i]}, i});
	}
	if(n <= 5000) {
			ll dp[m + 1][s + 1], pd[m + 1][s + 1];
	for(int j = 0; j <= m; ++j) 
		for(int k = 0; k <= s; ++k)
			dp[j][k] = pd[j][k] = -INF;
	dp[0][0] = 0;
	for(int i = 1; i <= n; ++i) {
		for(int j = 0; j <= m; ++j) {
			for(int k = 0; k <= s; ++k) {
				pd[j][k] = max({dp[j][k], (j == 0 ? -INF : dp[j - 1][k]  + a[i]), (k == 0 ? -INF : dp[j][k - 1] + b[i])});
			}
		}
		for(int j = 0; j <= m; ++j) {
			for(int k = 0; k <= s; ++k) {
		    	dp[j][k] = pd[j][k];
				pd[j][k] = -INF;
			}
		}
		dp[0][0] = 0;
	}
	
	cout << dp[m][s] << nl;  
		return;
	}
	sort(rall(v));
	ll A = 0, B = 0;
	for(int i = 0; i < m; ++i) {
		A += v[i].x.x;
		used[v[i].y] = 1;
	}        
	v.clear();
	for(int i = 1; i <= n; ++i) {
		if(used[i]) continue;
		v.pb({{b[i], 0}, 0});
	}
	sort(rall(v));
	for(int i = 0; i < s; ++i)
		A += v[i].x.x;

	v.clear();
	memset(used, 0, sizeof(used));
	
	for(int i = 1; i <= n; ++i) {
		v.pb({{b[i], -a[i]}, i});
	}
	sort(rall(v));
	for(int i = 0; i < s; ++i) {
		B += v[i].x.x;
		used[v[i].y] = 1;
	}
	v.clear();
	for(int i = 1; i <= n; ++i) {
		if(used[i]) continue;
		v.pb({{a[i], 0}, 0});
	}
	sort(rall(v));
	for(int i = 0; i < m; ++i)
		B += v[i].x.x;

	cout << max(A, B) << nl;
}
 
int main() {
	ios_base::sync_with_stdio(NULL);
	cin.tie(0);
	cout.tie(0);
	int test = 1;
	//cin >> test;
	while(test--) {
		solve();
	}
	return 0;
}

• Subtask 1: 35 / 100