CF 742 E

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值