水了一道紫题,因为做过加强版。
这题把两个区间改成一个,就可以直接统计最小值、最小值出现次数。
查询的时候特判一下最小值是否 $\leq 1$,是则累加答案。
这题是可以允许区间长度为 $1$ 的,所以不需要减。
#include <bits/stdc++.h>
using namespace std;
const int N = 3e5 + 15;
int n, a[N], pos[N];
bool st[N];
long long ans = 0;
struct Tree {
int l, r;
int Min; //最小
int cnt, flag;
} tr[N << 2];
void pushup(int u) {
tr[u].Min = min(tr[u << 1].Min, tr[u << 1 | 1].Min);
tr[u].cnt = 0;
if (tr[u << 1].Min == tr[u].Min) tr[u].cnt += tr[u << 1].cnt;
if (tr[u << 1 | 1].Min == tr[u].Min) tr[u].cnt += tr[u << 1 | 1].cnt;
}
void pushdown(int u) {
if (tr[u].flag) {
tr[u << 1].flag += tr[u].flag;
tr[u << 1 | 1].flag += tr[u].flag;
tr[u << 1].Min += tr[u].flag;
tr[u << 1 | 1].Min += tr[u].flag;
tr[u].flag = 0;
}
}
void build(int u, int l, int r) {
tr[u].l = l, tr[u].r = r;
if (l == r) {
tr[u].Min = 0;
tr[u].cnt = 1;
return;
}
int mid = l + r >> 1;
build(u << 1, l, mid);
build(u << 1 | 1, mid + 1, r);
pushup(u);
}
void change(int u, int l, int r, int d) {
if (r < tr[u].l || l > tr[u].r) return;
if (tr[u].l >= l && tr[u].r <= r) {
tr[u].Min += d, tr[u].flag += d;
return;
}
pushdown(u);
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) change(u << 1, l, r, d);
if (r > mid) change(u << 1 | 1, l, r, d);
pushup(u);
}
int query(int u, int l, int r) {
if (r < tr[u].l || l > tr[u].r) return 0;
if (tr[u].l >= l && tr[u].r <= r) {
if (tr[u].Min <= 1) return tr[u].cnt;
else return 0;
}
pushdown(u);
int mid = tr[u].l + tr[u].r >> 1, res = 0;
if (l <= mid) res += query(u << 1, l, r);
if (r > mid) res += query(u << 1 | 1, l, r);
return res;
}
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]), pos[a[i]] = i;
build(1, 1, n);
for (int i = 1; i <= n; i++) {
change(1, 1, i, 1);
if (st[pos[i] - 1]) change(1, 1, a[pos[i] - 1], -1);
if (st[pos[i] + 1]) change(1, 1, a[pos[i] + 1], -1);
st[pos[i]] = 1;
ans += query(1, 1, i); //这里就不需要减去一段了
}
printf("%lld\n", ans);
return 0;
}
背着我偷偷内卷是吧