hdu4027 Can you answer these queries?（线段树）

题目链接：点击链接

题意：

• 有n个数，现在有一种操作，将这n个数中的一个区间[x, y]，每个数都开平方。询问区间的和。

思路：

建树的时候，当前节点保存了对应区间的和。更新的时候，若当前节点对应的区间为[l, r]，如果区间和小于(r-l+1)，不往下更新，反之往下更新。

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>

using namespace std;

const int maxn = 100000 + 10;
typedef long long ll;
ll tree[maxn<<2];

void pushup(int rt){
    tree[rt] = tree[rt<<1] + tree[rt<<1|1];
}

void build(int rt, int l, int r){
    if(l==r){
        scanf("%lld", &tree[rt]);
    }
    else{
        int mid = (l+r) / 2;
        build(rt<<1, l, mid);
        build(rt<<1|1, mid+1, r);
        pushup(rt);
    }
}

void update(int rt, int a, int b, int l, int r){
    int mid = (l+r)>>1;
    if(a<=l && r<=b){
        if(l==r){
            tree[rt] = (ll)sqrt(tree[rt]);
            return;
        }
        if(tree[rt]>(r-l+1)){
            if(a <= mid) update(rt<<1, a, b, l, mid);
            if(mid < b) update(rt<<1|1, a, b, mid+1, r);
            pushup(rt);
        }
    }
    else{
        if(a <= mid) update(rt<<1, a, b, l, mid);
        if(mid < b) update(rt<<1|1, a, b, mid+1, r);
        pushup(rt);
    }
}

ll query(int rt, int a, int b, int l, int r){
    if(a<=l && r<=b){
        return tree[rt];
    }
    else{
        int mid = (l+r)>>1;
        ll ans = 0;
        if(a<=mid) ans += query(rt<<1, a, b, l, mid);
        if(mid<b) ans += query(rt<<1|1, a, b, mid+1, r);
        return ans;
    }
}

int main(){
    int n, q, opt, a, b, Case=0;
    while(scanf("%d", &n)!=EOF){
        build(1, 1, n);
        scanf("%d", &q);
        printf("Case #%d:\n", ++Case);
        for(int i=0; i<q; ++i){
            scanf("%d%d%d", &opt, &a, &b);
            if(opt==0){
                if(a>b) swap(a, b);
                update(1, a, b, 1, n);
            }
            if(opt==1){
                if(a>b) swap(a, b);
                printf("%lld\n", query(1, a, b, 1, n));
            }
        }
        printf("\n");
    }

    return 0;
}