답안 #954731

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
954731 2024-03-28T12:49:09 Z starchan 학교 설립 (IZhO13_school) C++17
100 / 100
102 ms 16588 KB
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pq priority_queue
#define in pair<int, int>
#define f first
#define s second
#define pb push_back
#define pob pop_back
#define INF (int)1e17
#define MX (int)3e5+5
#define fast() ios_base::sync_with_stdio(false); cin.tie(NULL)
struct ritul
{
	int Limit;
	int sz;
	int sum;
	pq<int> chungus;
	void init(int x)
	{
		Limit = x;
		sz = 0;
		while(!chungus.empty())
			chungus.pop();
		sum = 0;
		return;
	}
	void push(int x)
	{
		chungus.push(-x);
		sz++;
		sum+=x;
		if(sz > Limit)
		{
			sum+=chungus.top();
			sz--;
			chungus.pop();
		}
		return;
	}
	int ans()
	{
		return sum;
	}
};
signed main()
{
	fast(); ritul work1, work2;
	int n, A, B; cin >> n >> A >> B;
	vector<int> a(n+1), b(n+1), d(n+1, 0), e(n+1, 0); vector<in> c(n+1);
	for(int i = 1; i <= n; i++)
	{
		cin >> a[i] >> b[i];
		c[i] = {b[i]-a[i], i};
	}
	if((A == 0) && (B == 0))
	{
		cout << "0\n";
		return 0;
	}
	else if((A == 0))
	{
		b[0] = INF;
		sort(b.rbegin(), b.rend());
		int ans = 0;
		for(int i = 1; i <= B; i++)
			ans+=b[i];
		cout << ans << "\n";
		return 0;
	}
	else if((B == 0))
	{
		a[0] = INF;
		sort(a.rbegin(), a.rend());
		int ans = 0;
		for(int i = 1; i <= A; i++)
			ans+=a[i];
		cout << ans << "\n";
		return 0;
	}
	c[0] = {-INF, -INF};
	sort(c.begin(), c.end());
	work1.init(A);
	for(int i = 1; i <= n; i++)
	{
		work1.push(a[c[i].s]);
		d[i] = work1.ans();
	}
	work2.init(B);
	for(int i = n; i >= 1; i--)
	{
		work2.push(b[c[i].s]);
		e[i] = work2.ans();
	}
	int ans = -1;
	for(int i = A; i <= n-B; i++)
		ans = max(ans, d[i]+e[i+1]);
	cout << ans << "\n";
	return 0;
}			
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 604 KB Output is correct
8 Correct 2 ms 604 KB Output is correct
9 Correct 1 ms 604 KB Output is correct
10 Correct 2 ms 600 KB Output is correct
11 Correct 2 ms 600 KB Output is correct
12 Correct 2 ms 600 KB Output is correct
13 Correct 10 ms 2528 KB Output is correct
14 Correct 24 ms 4436 KB Output is correct
15 Correct 44 ms 7768 KB Output is correct
16 Correct 54 ms 11084 KB Output is correct
17 Correct 72 ms 12900 KB Output is correct
18 Correct 82 ms 13772 KB Output is correct
19 Correct 91 ms 14808 KB Output is correct
20 Correct 102 ms 16588 KB Output is correct