제출 #1146425

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1146425		monkey133	Bikeparking (EGOI24_bikeparking)	C++20	100 / 100	245 ms	23908 KiB

Main

#include <bits/stdc++.h>
//#include "includeall.h"
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define endl '\n'
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define lb lower_bound
#define ub upper_bound
#define input(x) scanf("%lld", &x);
#define input2(x, y) scanf("%lld%lld", &x, &y);
#define input3(x, y, z) scanf("%lld%lld%lld", &x, &y, &z);
#define input4(x, y, z, a) scanf("%lld%lld%lld%lld", &x, &y, &z, &a);
#define print(x, y) printf("%lld%c", x, y);
#define show(x) cerr << #x << " is " << x << endl;
#define show2(x,y) cerr << #x << " is " << x << " " << #y << " is " << y << endl;
#define show3(x,y,z) cerr << #x << " is " << x << " " << #y << " is " << y << " " << #z << " is " << z << endl;
#define all(x) x.begin(), x.end()
#define discretize(x) sort(x.begin(), x.end()); x.erase(unique(x.begin(), x.end()), x.end());
#define FOR(i, x, n) for (ll i =x; i<=n; ++i) 
#define RFOR(i, x, n) for (ll i =x; i>=n; --i) 
using namespace std;
mt19937_64 rnd(chrono::steady_clock::now().time_since_epoch().count());
//using namespace __gnu_pbds;
//#define ordered_set tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update>
//#define ordered_multiset tree<int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update>
typedef long long ll;
typedef long double ld;
typedef pair<ld, ll> pd;
typedef pair<string, ll> psl;
typedef pair<ll, ll> pi;
typedef pair<pi, ll> pii;
typedef pair<pi, pi> piii;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());



int main()
{
	ll n, tot = 0;
	cin >> n;
	vector<ll> have(n), want(n);
	set<pi> ss;
	for (ll i=0; i<n; ++i) {cin >> have[i]; ss.insert(mp(i, have[i]));}
	for (ll i=0; i<n; ++i) {cin >> want[i]; tot += want[i];}
	ll ans = 0;
	for (ll i=0; i<n; i++)
	{
		while (want[i])
		{
			auto it = ss.lb(mp(i, -1));
			if (it!=ss.begin()) it--;
			if (it->f >= i) break;
			pi now = *it;
			ss.erase(it);
			ll take = min(now.s, want[i]);
			want[i] -= take;
			now.s -= take;
			ans += 2*take;
			if (now.s) ss.insert(now);
			
		}
	}
	for (ll i=0; i<n; i++)
	{
		while (want[i])
		{
			auto it = ss.lb(mp(i, -1));
			if (it!=ss.begin()) it--;
			if (it->f > i) break;
			pi now = *it;
			ss.erase(it);
			ll take = min(now.s, want[i]);
			want[i] -= take;
			now.s -= take;
			ans += take;
			if (now.s) ss.insert(now);
			
		}
	}
	cout << ans - tot << endl;
	
	return 0;
}

• Subtask 1: 16 / 16
• Subtask 2: 9 / 9
• Subtask 3: 19 / 19
• Subtask 4: 24 / 24
• Subtask 5: 32 / 32