Submission #506013

#	Time	Username	Problem	Language	Result	Execution time	Memory
506013		sidon	Candies (JOI18_candies)	C++17	100 / 100	196 ms	9416 KiB

candies

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define Z 200'002

int N, K, L[Z], R[Z], r[Z];
ll A[Z], sum;
priority_queue<array<ll, 2>> pq;

int main() {
	cin >> N;

	for(int i = 1; i <= N; i++) {
		cin >> A[i];
		L[i] = i - 1, R[i] = i + 1;
		pq.push({A[i], i});
	}
 
	R[N+1] = N+1, L[N+1] = N;
	R[0] = r[0] = r[N+1] = 1;
 
	for(int k = 1; k*2 <= N+1; k++) {
		while(r[pq.top()[1]]) pq.pop();
		auto [v, i] = pq.top(); pq.pop();
 
		cout << (sum += v) << '\n';
		r[i] = r[L[i]] || r[R[i]];
		if(!r[L[i]]) v -= A[L[i]], r[L[i]] = 1, R[L[i] = L[L[i]]] = i;
		if(!r[R[i]]) v -= A[R[i]], r[R[i]] = 1, L[R[i] = R[R[i]]] = i;

		if(!r[i]) pq.push({A[i] = -v, i});
	}
}

• Subtask 1: 8 / 8
• Subtask 2: 92 / 92