hdu 4027 Can you answer these queries? 线段树

题目链接

给出n个数m个操作， 每个操作有两种， 1是查询[L, R]的和， 2是将[L, R]之间的数变为他们的平方根， 4变成2， 9变成3， 如果不是整数就向下取整， 数据<2^63。

如果一个数是1， 那么它不会变， 一个数不是1， 即使它非常大， 也可以在很少的操作之中变成1。 所以直接更新到叶子节点， 然后用一个数组记录这一个范围用不用更新， 如果全都是1就可以不更新。

另外这题操作时给的范围x, y， x貌似有可能大于y。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define lson l, m, rt<<1
#define mem(a) memset(a, 0, sizeof(a))
#define rson m+1, r, rt<<1|1
const int maxn = 1e5+5;
ll sum[maxn<<2], cnt[maxn<<2];    //cnt数组记录一个范围是否需要更新
void pushUp(int rt) {
    sum[rt] = sum[rt<<1]+sum[rt<<1|1];
    cnt[rt] = cnt[rt<<1]&cnt[rt<<1|1];
}
void build(int l, int r, int rt) {
    if(l == r) {
        scanf("%I64d", &sum[rt]);
        if(sum[rt]<=1)
            cnt[rt] = 1;
        return ;
    }
    int m = l+r>>1;
    build(lson);
    build(rson);
    pushUp(rt);
}
void update(int L, int R, int l, int r, int rt) {
    if(l == r) {
        sum[rt] = int(sqrt(sum[rt]*1.0));
        if(sum[rt]<=1)
            cnt[rt] = 1;
        return ;
    }
    int m = l+r>>1;
    if(L<=m&&!cnt[rt<<1])
        update(L, R, lson);
    if(R>m&&!cnt[rt<<1|1])
        update(L, R, rson);
    pushUp(rt);
}
ll query(int L, int R, int l, int r, int rt) {
    if(L<=l&&R>=r) {
        return sum[rt];
    }
    int m = l+r>>1;
    ll ret = 0;
    if(L<=m)
        ret += query(L, R, lson);
    if(R>m)
        ret += query(L, R, rson);
    return ret;
}
int main()
{
    int n, m, num = 1;;
    while(cin>>n) {
        mem(cnt);
        build(1, n, 1);
        cin>>m;
        printf("Case #%d:\n", num++);
        while(m--) {
            int sign, x, y;
            scanf("%d%d%d", &sign, &x, &y);
            if(x>y)
                swap(x, y);
            if(sign) {
                printf("%I64d\n", query(x, y, 1, n, 1));
            } else {
                update(x, y, 1, n, 1);
            }
        }
        cout<<endl;
    }
}