oj.uz

Submission #497277

# Submission time Handle Problem Language Result Execution time Memory
497277 2021-12-22T22:39:27 Z aryan12 Growing Trees (BOI11_grow) C++17
100 / 100
 189 ms 18568 KB 
#include <bits/stdc++.h>
using namespace std;
#define int long long

mt19937_64 RNG(chrono::steady_clock::now().time_since_epoch().count());

const int N = 1e5 + 5, INF = 1e15;
vector<int> a(N);
int n, q;

struct range {
	int minval, maxval, lazyval;

	range() {
		minval = 0;
		maxval = 0;
		lazyval = 0;
	}
} *seg[N * 4];

void FixLazy(int l, int r, int pos) {
	seg[pos]->maxval += seg[pos]->lazyval;
	seg[pos]->minval += seg[pos]->lazyval;
	if(l == r) {
		seg[pos]->lazyval = 0;
		return;
	}
	seg[pos * 2]->lazyval += seg[pos]->lazyval;
	seg[pos * 2 + 1]->lazyval += seg[pos]->lazyval;
	seg[pos]->lazyval = 0;
}

void Upd(int l, int r, int pos, int ql, int qr, int qval) {
	if(seg[pos]->lazyval != 0) {
		FixLazy(l, r, pos);
	}
	if(ql > r || l > qr) {
		return;
	}
	if(l == r) {
		seg[pos]->minval += qval;
		seg[pos]->maxval += qval;
		return;
	}
	if(ql <= l && r <= qr) {
		seg[pos]->minval += qval;
		seg[pos]->maxval += qval;
		seg[pos * 2]->lazyval += qval;
		seg[pos * 2 + 1]->lazyval += qval;
		return;
	}
	int mid = (l + r) >> 1;
	Upd(l, mid, pos * 2, ql, qr, qval);
	Upd(mid + 1, r, pos * 2 + 1, ql, qr, qval);
	seg[pos]->minval = min(seg[pos * 2]->minval, seg[pos * 2 + 1]->minval);
	seg[pos]->maxval = max(seg[pos * 2]->maxval, seg[pos * 2 + 1]->maxval);
}

int GetValue(int l, int r, int pos, int qpos) {
	if(seg[pos]->lazyval != 0) {
		FixLazy(l, r, pos);
	}
	if(l == r) {
		return seg[pos]->maxval;
	}
	int mid = (l + r) >> 1;
	if(qpos <= mid) {
		return GetValue(l, mid, pos * 2, qpos);
	}
	return GetValue(mid + 1, r, pos * 2 + 1, qpos);
}

int FindIdx(int l, int r, int pos, int qval) {
	if(seg[pos]->lazyval != 0) {
		FixLazy(l, r, pos);
	}
	if(seg[pos]->maxval <= qval) {
		return r;
	}
	if(seg[pos]->minval > qval) {
		return l - 1;
	}
	int mid = (l + r) >> 1;
	int x = FindIdx(l, mid, pos * 2, qval);
	if(x == mid) {
		return FindIdx(mid + 1, r, pos * 2 + 1, qval);
	}
	return x;
}

void Output(int l, int r, int pos) {
	if(seg[pos]->lazyval != 0) {
		FixLazy(l, r, pos);
	}
	if(l == r) {
		cout << seg[pos]->maxval << " ";
		return;
	}
	int mid = (l + r) >> 1;
	Output(l, mid, pos * 2);
	Output(mid + 1, r, pos * 2 + 1);
}

void Solve() {
	for(int i = 0; i < N * 4; i++) {
		seg[i] = new range();
	}
	cin >> n >> q;
	for(int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	sort(a.begin() + 1, a.begin() + 1 + n);
	for(int i = 1; i <= n; i++) {
		//cout << "a[i] = " << a[i] << "\n";
		Upd(1, n, 1, i, i, a[i]);
	}
	for(int i = 1; i <= q; i++) {
		char command;
		cin >> command;
		if(command == 'F') {
			int freq, atleast;
			cin >> freq >> atleast;
			if(seg[1]->maxval < atleast) {
				continue;
			}
			int startfrom = FindIdx(1, n, 1, atleast - 1) + 1;
			//cout << "How many trees? " << freq << ", Min height? " << atleast << "\n";
			//cout << "startfrom = " << startfrom << "\n";
			int lastPos = min(n, startfrom + freq - 1);
			if(lastPos == n) {
				Upd(1, n, 1, startfrom, lastPos, 1);
				continue;
			}
			//cout << "lastPos = " << lastPos << "\n";
			int lastValue = GetValue(1, n, 1, lastPos);
			//cout << "lastValue = " << lastValue << "\n";
			int changesRequired = FindIdx(1, n, 1, lastValue - 1);
			//cout << "changesRequired = " << changesRequired << "\n";
			Upd(1, n, 1, startfrom, changesRequired, 1);
			int freqLeft = freq - (changesRequired - startfrom + 1);
			//cout << "freqLeft = " << freqLeft << "\n";
			lastPos = FindIdx(1, n, 1, lastValue);
			//cout << "lastPos = " << lastPos << "\n";
			Upd(1, n, 1, lastPos - freqLeft + 1, lastPos, 1);
		}
		else {
			int minx, maxx;
			cin >> minx >> maxx;
			cout << FindIdx(1, n, 1, maxx) - FindIdx(1, n, 1, minx - 1) << "\n";
		}
		//cout << "Seg tree at the end looks like:\n";
		//Output(1, n, 1);
		//cout << "\n";
	}
}

int32_t main() {
	auto begin = std::chrono::high_resolution_clock::now();
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int t = 1;
	//cin >> t;
	while(t--) {
		Solve();
	}
	auto end = std::chrono::high_resolution_clock::now();
    auto elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - begin);
    cerr << "Time measured: " << elapsed.count() * 1e-9 << " seconds.\n"; 
	return 0;
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 99 ms 17976 KB Output is correct
2 Correct 157 ms 18436 KB Output is correct
3 Correct 60 ms 18456 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 16 ms 16716 KB Output is correct
2 Correct 16 ms 16784 KB Output is correct
3 Correct 15 ms 16784 KB Output is correct
4 Correct 14 ms 16780 KB Output is correct
5 Correct 54 ms 17800 KB Output is correct
6 Correct 62 ms 17964 KB Output is correct
7 Correct 18 ms 16776 KB Output is correct
8 Correct 31 ms 17564 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 53 ms 18016 KB Output is correct
2 Correct 63 ms 18172 KB Output is correct
3 Correct 15 ms 16720 KB Output is correct
4 Correct 38 ms 17672 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 47 ms 18104 KB Output is correct
2 Correct 66 ms 18084 KB Output is correct
3 Correct 23 ms 16848 KB Output is correct
4 Correct 62 ms 18140 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 91 ms 17776 KB Output is correct
2 Correct 146 ms 18108 KB Output is correct
3 Correct 30 ms 17132 KB Output is correct
4 Correct 55 ms 18156 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 131 ms 17744 KB Output is correct
2 Correct 148 ms 18304 KB Output is correct
3 Correct 63 ms 18404 KB Output is correct
4 Correct 30 ms 17080 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 93 ms 17920 KB Output is correct
2 Correct 104 ms 18020 KB Output is correct
3 Correct 60 ms 18400 KB Output is correct
4 Correct 30 ms 17148 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 155 ms 17892 KB Output is correct
2 Correct 143 ms 18108 KB Output is correct
3 Correct 61 ms 17460 KB Output is correct
4 Correct 70 ms 17928 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 124 ms 18024 KB Output is correct
2 Correct 148 ms 18496 KB Output is correct
3 Correct 189 ms 18568 KB Output is correct

Subtask #10
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 135 ms 18296 KB Output is correct