/*
* Similar to editorial, but I just use one segment tree that supports updates and range max.
* To find "longest sequence such that all elements are less that d starting at a" on either side of a, I use binary search.
* This does make it log^2(n), but passes easily.
*/
#include <bits/stdc++.h>
using namespace std;
#define for_(i, s, e) for (int i = s; i < (int) e; i++)
#define for__(i, s, e) for (ll i = s; i < e; i++)
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> ii;
#define endl '\n'
class SegmentTree {
private:
vi tree, raw; int n;
int merge(int a, int b) {
return max(a, b);
}
void buildTree(int l, int r, int p) {
if (l == r) {
tree[p] = raw[l];
return;
}
int mid = (l + r) / 2;
int c1 = 2*p+1, c2 = 2*p+2;
buildTree(l, mid, c1); buildTree(mid+1, r, c2);
tree[p] = merge(tree[c1], tree[c2]);
}
void update(int i, int val, int l, int r, int p) {
if (l > i or r < i) return;
if (l == r) {
tree[p] = val;
return;
}
int mid = (l + r) / 2;
int c1 = 2*p+1, c2 = 2*p+2;
update(i, val, l, mid, c1); update(i, val, mid+1, r, c2);
tree[p] = merge(tree[c1], tree[c2]);
}
int mx(int i, int j, int l, int r, int p) {
if (l > j or r < i) return 0;
if (l >= i and r <= j) return tree[p];
int mid = (l + r) / 2;
int c1 = 2*p+1, c2 = 2*p+2;
return merge(mx(i, j, l, mid, c1), mx(i, j, mid+1, r, c2));
}
public:
SegmentTree(vi input) {
raw = input;
n = raw.size();
tree.resize(4*n);
buildTree(0, n-1, 0);
}
int mx(int i, int j) {
return mx(i, j, 0, n-1, 0);
}
void update(int i, int val) {
update(i, val, 0, n-1, 0);
}
};
int main() {
#ifdef shiven
freopen("test.in", "r", stdin);
#endif
ios_base::sync_with_stdio(false);
cin.tie(0);
int n, a, q; cin >> n >> a;
a -= 1;
vi nums(n);
for_(i, 0, n) {
cin >> nums[i];
pos[nums[i]] = i;
}
SegmentTree tree(nums);
vector<ii> mod(n);
for_(i, 0, n) mod[i] = {nums[i], i};
sort(mod.begin(), mod.end(), greater<ii>());
set<ii, greater<ii>> s;
for_(i, 0, min(10, n)) s.insert({mod[i].first, mod[i].second});
cin >> q;
while (q--) {
char c; cin >> c;
if (c == 'F') {
int k; cin >> k;
k -= 1;
if (k == a) {
cout << 0 << endl;
continue;
}
int bef = abs(k-a), lo, hi, ans, largest;
if (k < a) {
largest = tree.mx(k, a-1);
lo = a+1, hi = ans = n;
while (lo < hi) {
int mid = (lo+hi)/2;
if (tree.mx(a+1, mid) >= largest) {
ans = hi = mid;
} else {
lo = mid+1;
}
}
} else {
largest = tree.mx(a+1, k);
lo = 0, hi = a, ans = -1;
while (lo < hi) {
int mid = (lo+hi)/2;
if (tree.mx(mid, a-1) >= largest) {
ans = mid;
lo = mid+1;
} else {
hi = mid;
}
}
}
cout << bef+abs(ans-a)-1 << endl;
} else {
int x, y; cin >> x >> y;
set<ii, greater<ii>> ss;
x -= 1; y -= 1;
int pt = 0, nv = -1;
for (auto i: s) {
if (ss.size()+1 == s.size()) break;
else if (pt >= y) {
if (i.second != x) {
if (nv == -1) nv = i.first+1;
ss.insert(i);
}
}
else {
tree.update(i.second, i.first+1);
nums[i.second] += 1;
ss.insert({i.first+1, i.second});
nv = i.first;
}
pt++;
}
tree.update(x, nv);
ss.insert({nv, x});
swap(s, ss);
}
}
return 0;
}
Compilation message
cake.cpp: In function 'int main()':
cake.cpp:88:3: error: 'pos' was not declared in this scope; did you mean 'pow'?
88 | pos[nums[i]] = i;
| ^~~
| pow
