hdu 6315 Naive Operations

最新推荐文章于 2019-10-30 17:45:26 发布

原创最新推荐文章于 2019-10-30 17:45:26 发布 · 177 阅读

0 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用线段树解决区间更新与查询问题的方法。通过维护区间内的最小值和累加和，在进行区间更新时能高效地计算出特定条件下的累加和结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

开始给定两个长度为n的序列， b是1到n， a全是0

两个操作：1. add a[l] ~a[r] 均加一 2. 查询区间l~r的 a[i]/b[i]的和

令a[i]初始化为b[i],线段树维护一下a数组, 每次add操作区间减一

显然每次当a[i]变成0时，这位数的a[i]/b[i] 加一，维护一个答案区间求和，当update操作后要搜一下有多少个点变成了0，然后将其恢复成b[i]

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ms(x) memset(x, 0, sizeof(x))
using namespace std;

#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int MAXN = 1e5+3;
const int INF = 0x3f3f3f3f;
int b[MAXN<<2];
int SUM[MAXN<<2], MIN[MAXN<<2];
int lazy[MAXN<<2];

void PushUp(int rt) {
    SUM[rt] = SUM[rt<<1] + SUM[rt<<1|1];
    MIN[rt] = min(MIN[rt<<1], MIN[rt<<1|1]);
}

void PushDown(int rt) {
    if(lazy[rt]) {
        lazy[rt<<1] += lazy[rt];
        lazy[rt<<1|1] += lazy[rt];
        MIN[rt<<1] -= lazy[rt];
        MIN[rt<<1|1] -= lazy[rt];
        lazy[rt] = 0;
    }
}

// 以下的 l, r, rt 带入 1, n, 1
void Build(int l, int r, int rt) {
    lazy[rt] = 0;
    if(l == r) {
        // 初始化树为 0 的写法
        MIN[rt] = b[l];
        SUM[rt] = 0;
        return;
    }
    int m = (l + r) >> 1;
    Build(lson);
    Build(rson);
    PushUp(rt);
}

// 将区间 L~R 的值增加 v
void Update(int L, int R, int l, int r, int rt) {
    if(L<=l && r<=R) {
        lazy[rt] += 1;
        MIN[rt] -= 1;
        return;
    }
    PushDown(rt);
    int m = (l + r) >> 1;
    if(L <= m) Update(L, R,  lson);
    if(R > m) Update(L, R, rson);
    PushUp(rt);
}
void dfs(int l, int r, int rt){
    if(MIN[rt]!=0) return ;
    if(l==r){
        MIN[rt] = b[l];
        SUM[rt]++;
        return ;
    }
    PushDown(rt);
    int m = (l + r) >> 1;
    dfs(lson);
    dfs(rson);
    PushUp(rt);
}

// 求区间 L~R 的和
int QuerySum(int L, int R, int l, int r, int rt) {
    if(L<=l && r<=R) return SUM[rt];
    PushDown(rt);
    int m = (l + r) >> 1;
    int ret = 0;
    if(L <= m) ret += QuerySum(L, R, lson);
    if(R > m) ret += QuerySum(L, R, rson);

    return ret;
}

// 求区间 L~R 的最小值
int QueryMin(int L, int R, int l, int r, int rt) {
    if(L<=l && r<=R) return MIN[rt];
    PushDown(rt);
    int m = (l + r) >> 1;
    int ret = INF;
    if(L <= m) ret = min(ret, QueryMin(L, R, lson));
    if(R > m) ret = min(ret, QueryMin(L, R, rson));

    return ret;
}

int main(){
    int n, m;
    while(scanf("%d%d", &n, &m)!=EOF){
    ms(lazy);
    for(int i=1;i<=n;i++){
        scanf("%d", &b[i]);
    }
    Build(1,n,1);
    while(m--){
        char s[10];
        int l, r;
        scanf("%s%d%d",s, &l, &r);
        if(s[0] == 'a'){
            Update(l,r,1,n,1);
            dfs(1,n,1);
        }
        else{
            printf("%d\n", QuerySum(l,r,1,n,1));
        }
    }
    }
    return 0;
}