acmoj - 数蘑菇线段树区间更新

数蘑菇

题目描述

某土豪有许多种蘑菇的房子（编号从1到n，n<=10000）,每个房子开始时有个蘑菇，大小在2到1000之间，每当洒脱的土豪给他们浇水时，这些蘑菇就会变成原来的平方倍大，比如一个大小为2的蘑菇在土豪浇过水之后就会变成4这么大，但是因为土豪住在帝都，每个房子只有1000平方米这么大，所以每个蘑菇在长到1000之后就不会再长大了（比如对于一个大小为50的蘑菇浇水，它也只能长大到1000），土豪每天喜欢骑着自行车（乡下人羡慕城里人开车溜达，城里的土豪却骑着自行车，你们城里人真会玩）给一连串的房子里的蘑菇浇水，从l到r，然后土豪又经常要查询某些房间的蘑菇一共有多少了，好联系卖家，聪明的你，能帮土豪这个忙吗？搞不好土豪一高兴随便送套房子给你呢！

输入描述

输入首先是一个T，表示一共有T组数据

下一行是一个n，代表土豪有几间房子

下一行是n个整数数，代表从1到n每个房间开始蘑菇的大小(<=1000)

下一行是一个整数k(1<=k<=10000),表示土豪浇水和数蘑菇的次数，

后面跟着k行，每行三个整数q,l,r（1<=l<=r<=10000）

q表示土豪的操作，

q为0的时候表示土豪给l到r这些房间里的蘑菇浇水，

q为1的时候表示土豪想要知道l到到r这些房间里的蘑菇总大小

输出描述

对于第n组数据首先输出一行"Case #n:"，然后对于每次询问输出一个答案，每个答案占一行

输入样例

1
10
2 3 4 5 6 7 8 9 10 11
3
0 1 10
1 1 10
1 1 5

输出样例

Case #1:
505
90

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 10000 + 5;
int n, T, m;
int sum[maxn << 2],col[maxn << 2];
bool isupd[maxn << 2];//记录着他的子区间是否有要更新的
void inline pushup(int rt) {
    sum[rt] = sum[rt << 1] + sum[rt << 1|1];
    if(rt & 1)
        isupd[(rt + 1) / 2] = isupd[rt];//如果子有要更新的，就要标记父亲节点，告诉他子节点有更新
    else
        isupd[rt / 2] = isupd[rt];
}
void inline pushdown(int rt) {
    if(col[rt]) {
        col[rt << 1] += col[rt];
        col[rt << 1|1] += col[rt];
        col[rt] = 0;
    }
    isupd[rt << 1] = isupd[rt];
    isupd[rt << 1|1] = isupd[rt];
}
void build(int rt, int l, int r) {
    col[rt] = 0;
    isupd[rt] = 0;
    if(l == r) {
        scanf("%d", &sum[rt]);
        return;
    }
    int mid = (l + r) >> 1;
    build(rt << 1, l, mid);
    build(rt << 1|1, mid + 1, r);
    pushup(rt);
}
void update(int L, int R, int c, int rt, int l, int r) {
    if(L <= l && r <= R) {
        col[rt] += c;
        isupd[rt] = 1;
        return;
    }
    pushdown(rt);
    int mid = (l + r) >> 1;
    if(L <= mid) update(L, R, c, rt << 1, l, mid);
    if(R > mid) update(L, R, c,rt << 1|1, mid + 1, r);
    pushup(rt);
}
int query(int L, int R, int rt, int l, int r) {
    if(l == r) {
        if(sum[rt] == 1000)return sum[rt];//必须剪枝，否则超时
        if(isupd[rt]) {
            for(int i = 0; i < col[rt]; i ++) {
                sum[rt] = sum[rt] * sum[rt];
                if(sum[rt] >= 1000) sum[rt] = 1000;
            }
            col[rt] = 0;
            isupd[rt] = 0;
        }
        return sum[rt];
    }
    if(L <= l && r <= R) {
        int res = sum[rt];
        if(sum[rt] == 1000 * (r - l + 1))return sum[rt];//必须剪枝，否则超时
        if(isupd[rt]) {//如果子节点有更新，那么他的值就要重新通过求和来确定
            res = 0;
            int mid = (l + r) >> 1;
            pushdown(rt);
            res += query(l, r, rt << 1 , l, mid);
            res += query(l, r, rt << 1|1, mid + 1, r);
            isupd[rt] = 0;
        }
        return sum[rt] = res;
    }
    int mid = (l + r) >> 1;
    int res = 0;
    if(L <= mid)res += query(L, R, rt << 1, l, mid);
    if(R > mid) res += query(L, R, rt << 1|1, mid + 1, r);
    pushup(rt);
    return res;
}
int main() {
    int q, l, r, tt = 1;
    //freopen("D://imput.txt","r",stdin);
    scanf("%d", &T);
    while(T --) {
        scanf("%d", &n);
        build(1, 1, n);
        scanf("%d", &m);
        printf("Case #%d:\n",tt ++);
        for(int i = 0; i < m; i ++) {
            scanf("%d%d%d", &q, &l, &r);
            if(q) {
                printf("%d\n",query(l, r, 1, 1, n));
            } else {
                update(l, r, 1, 1, 1, n);
            }
        }
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。