oj.uz

Submission #837689

# Submission time Handle Problem Language Result Execution time Memory
837689 2023-08-25T14:08:32 Z NothingXD Distributing Candies (IOI21_candies) C++17
67 / 100
 539 ms 38880 KB 
#include "candies.h"
#include <bits/stdc++.h>

//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx2")

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

void debug_out(){cerr << endl;}
template<typename Head, typename... Tail>
void debug_out(Head H, Tail... T){
	cerr << H << ' ';
	debug_out(T...);
}

#define debug(...) cerr << "(" << #__VA_ARGS__ << "): ", debug_out(__VA_ARGS__)
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define MP(x, y) make_pair(x, y)

const int maxn = 2e5 + 10;
const ll inf = 1e18;
int n, q, c[maxn];
vector<pii> Q[maxn];
pll seg[maxn << 2];
ll lazy[maxn << 2];

pll operator +(pll a, pll b){
	return MP(min(a.F, b.F), max(a.S, b.S));
}

void shift(int v, int l, int r);

void add(int v, int l, int r, int ql, int qr, ll val){
	if (qr <= l || r <= ql) return;
	if (ql <= l && r <= qr){
		seg[v].F += val;
		seg[v].S += val;
		lazy[v] += val;
		return;
	}
	shift(v, l, r);
	int mid = (l + r) / 2;
	add(v << 1, l, mid, ql, qr, val);
	add(v << 1 | 1, mid, r, ql, qr, val);
	seg[v] = seg[v << 1] + seg[v << 1 | 1];
}

int find(int v, int l, int r, int val, pll &suf){
	//debug(v, l, r, val, seg[v].S, suf.S, seg[v].F, suf.F);
	if (max(seg[v].S, suf.S) - min(seg[v].F, suf.F) < val){
		suf = suf + seg[v];
		return -1;
	}
	if (l + 1 == r){
		suf = suf + seg[v];
		return l;
	}
	shift(v, l, r);
	int mid = (l + r) / 2;
	int res = find(v << 1 | 1, mid, r, val, suf);
	if (res == -1) res = find(v << 1, l, mid, val, suf);
	return res;
}

pii find2(int v, int l, int r, int ql, int qr, pll val){
	if (qr <= l || r <= ql) return {-1, 0};
	if (ql <= l && r <= qr && seg[v].F > val.F && seg[v].S < val.S) return {-1, 0};
	if (l + 1 == r){
		pii ret = {l, 0};
		if (seg[v].S == val.S) ret.S = 1;
		return ret;
	}
	shift(v, l, r);
	int mid = (l + r) / 2;
	pii res = find2(v << 1, l, mid, ql, qr, val);
	if (res.F == -1) res = find2(v << 1 | 1, mid, r, ql, qr, val);
	return res;
}

int get(int v, int l, int r, int idx){
	if (l + 1 == r) return seg[v].F;
	shift(v, l, r);
	int mid = (l + r) / 2;
	if (idx < mid) return get(v << 1, l, mid, idx);
	return get(v << 1 | 1, mid, r, idx);
}

void shift(int v, int l, int r){
	if (!lazy[v]) return;
	int mid = (l + r) / 2;
	add(v << 1, l, mid, l, mid, lazy[v]);
	add(v << 1 | 1, mid, r, mid, r, lazy[v]);
	lazy[v] = 0;
}

vector<int> distribute_candies(vector<int> C, vector<int> L, vector<int> R, vector<int> V){
	n = C.size();
	q = L.size();
	for (int i = 0; i < n; i++){
		c[i] = C[i];
	}
	for (int i = 0; i < q; i++){
		Q[L[i]].push_back({i, V[i]});
		Q[R[i]+1].push_back({i, -V[i]});
	}
	add(1, -2, q, -2, -1, inf);
	vector<int> ans(n);
	for (int i = 0; i < n; i++){
		for (auto [x, y]: Q[i]){
	//		debug(x, y);
			add(1, -2, q, x, q, y);
		}
		pll tmp = {inf, -inf};
		int idx = find(1, -2, q, c[i], tmp);
	//	debug(i, idx, tmp.F, tmp.S);
		pii idx2 = find2(1, -2, q, idx+1, q, tmp);
	//	debug(idx2.F, idx2.S);
		ans[i] = get(1, -2, q, q) - get(1, -2, q, idx2.F);
	//	debug(ans[i]);
		if (idx2.S == 1) ans[i] += c[i];
	//	debug(ans[i]);
	}
	return ans;
}

Subtask #1
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 4 ms 5152 KB Output is correct
4 Correct 4 ms 5204 KB Output is correct
5 Correct 6 ms 5204 KB Output is correct

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 484 ms 32160 KB Output is correct
2 Correct 495 ms 36252 KB Output is correct
3 Correct 539 ms 35972 KB Output is correct

Subtask #3
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 269 ms 29528 KB Output is correct
3 Correct 131 ms 11732 KB Output is correct
4 Correct 429 ms 38104 KB Output is correct
5 Correct 481 ms 38496 KB Output is correct
6 Correct 431 ms 38880 KB Output is correct
7 Correct 433 ms 38216 KB Output is correct

Subtask #4
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 4 ms 4948 KB Output is correct
3 Correct 96 ms 27912 KB Output is correct
4 Correct 94 ms 9584 KB Output is correct
5 Correct 217 ms 31716 KB Output is correct
6 Correct 235 ms 32488 KB Output is correct
7 Correct 197 ms 33068 KB Output is correct
8 Correct 221 ms 31616 KB Output is correct
9 Correct 253 ms 33316 KB Output is correct

Subtask #5
0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 4 ms 5152 KB Output is correct
4 Correct 4 ms 5204 KB Output is correct
5 Correct 6 ms 5204 KB Output is correct
6 Correct 484 ms 32160 KB Output is correct
7 Correct 495 ms 36252 KB Output is correct
8 Correct 539 ms 35972 KB Output is correct
9 Correct 2 ms 4948 KB Output is correct
10 Correct 269 ms 29528 KB Output is correct
11 Correct 131 ms 11732 KB Output is correct
12 Correct 429 ms 38104 KB Output is correct
13 Correct 481 ms 38496 KB Output is correct
14 Correct 431 ms 38880 KB Output is correct
15 Correct 433 ms 38216 KB Output is correct
16 Correct 2 ms 4948 KB Output is correct
17 Correct 4 ms 4948 KB Output is correct
18 Correct 96 ms 27912 KB Output is correct
19 Correct 94 ms 9584 KB Output is correct
20 Correct 217 ms 31716 KB Output is correct
21 Correct 235 ms 32488 KB Output is correct
22 Correct 197 ms 33068 KB Output is correct
23 Correct 221 ms 31616 KB Output is correct
24 Correct 253 ms 33316 KB Output is correct
25 Incorrect 2 ms 4948 KB Output isn't correct
26 Halted 0 ms 0 KB -