CF 742 E
[link](Problem - E - Codeforces)
题意:
给定一个长度位n的序列和m次操作,操作1为将第x个数替换为y,操作二为输出[l, r] 区间内的非递减序列的数目。
题解:
用线段树来维护区间非递减的数目。假设当前节点为u, u的res一定是u的左儿子的res加u的右儿子的res以及是否左右儿子能够合并带来的贡献。左右儿子的res就是线段树维护的答案很好看出来,直接加上就可以。
现在单独来看合并带来的贡献,当且仅当左儿子的右端点小于等于右儿子的左端点时才可以合并。那么合并带来的贡献是什么呢?贡献即左儿子的右端点往前延申的最长符合非递减序列的长度,也就是最长后缀rm和右儿子的左端点往后延申的最长符合非递减序列的长度,也就是最长前缀lm,的乘积,因为假设后缀为[l, r] 那么 [l, r] 产生的贡献是1 * lm、[l + 1, r] 产生的贡献是1 * lm以此类推,符合乘法原理,所以合并贡献就是lm * rm。
所以经过一通分析,线段树需要维护就显而易见了
l, r区间
lm, rm 最长前后缀
sum 取前非递减序列的数目
最后每次看当前点的两个子节点是否能合并,往上pushup更新维护即可。
Code
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <set>
#include <queue>
#include <vector>
#include <map>
#include <unordered_map>
#include <cmath>
#include <stack>
#include <iomanip>
#include <deque>
#include <sstream>
#define x first
#define y second
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 2e5 + 10, M = 2 * N, INF = 0x3f3f3f3f, mod = 1e9 + 7;
const double eps = 1e-8;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
struct Node {
int l, r;
int lm, rm; // 前缀和后缀最长距离
LL sum;
} tr[N * 4];
int n, m;
int a[N];
void pushup(int u) {
tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum; // 左右子节点的和
tr[u].lm = tr[u << 1].lm, tr[u].rm = tr[u << 1 | 1].rm; //前缀和后缀
if (a[tr[u << 1].r] <= a[tr[u << 1 | 1].l]) {
if (tr[u << 1].lm == tr[u << 1].r - tr[u << 1].l + 1) tr[u].lm += tr[u << 1 | 1].lm; // 前缀延申
if (tr[u << 1 | 1].rm == tr[u << 1 | 1].r - tr[u << 1 | 1].l + 1) tr[u].rm += tr[u << 1].rm; //后缀延申
tr[u].sum += 1ll * tr[u << 1].rm * tr[u << 1 | 1].lm; // 连续延申贡献
}
}
void build(int u, int l, int r) {
tr[u] = {l, r};
if (l == r) {
tr[u].sum = 1, tr[u].lm = tr[u].rm = 1;
return ;
}
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
pushup(u);
}
void update(int u, int x, int v) {
int l = tr[u].l, r = tr[u].r;
if (l == r) {
a[x] = v;
return ;
}
int mid = (l + r) >> 1;
if (mid >= x) update(u << 1, x, v);
else update(u << 1 | 1, x, v);
pushup(u); // 更新当前节点信息
}
LL query(int u, int l, int r) {
if (l <= tr[u].l && tr[u].r <= r) return tr[u].sum;
int mid = tr[u].l + tr[u].r >> 1;
LL res = 0;
if (mid >= l) res += query(u << 1, l, r);
if (mid < r) res += query(u << 1 | 1, l, r);
if (a[tr[u << 1].r] <= a[tr[u << 1 | 1].l]) { // 合并,虽然当前区间不能完全包含,但是当前区间的mid被包含在要查询的[l, r]中了,那么他的左右区间能合并也会产生贡献,要加上。
int lsum = min(mid - l + 1, tr[u << 1].rm), rsum = min(r - mid, tr[u << 1 | 1].lm);
if (lsum > 0 && rsum > 0)res += 1ll * lsum * rsum; // 大于零有贡献再加
}
return res;
}
int main() {
ios::sync_with_stdio(false), cin.tie(0);
cin >> n >> m;
for (int i = 1; i <= n; i ++ ) cin >> a[i];
build(1, 1, n);
while (m --) {
int op;
cin >> op;
if (op == 1) {
int u, x;
cin >> u >> x;
update(1, u, x);
}
else {
int l, r;
cin >> l >> r;
cout << query(1, l, r) << endl;
}
}
return 0;
}