[BZOJ3038]上帝造题的七分钟2

上帝造题的七分钟2

Description
XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。
“第一分钟，X说，要有数列，于是便给定了一个正整数数列。
第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。
第三分钟，k说，要能查询，于是便有了求一段数的和的操作。
第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。
第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。
第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。”
——《上帝造题的七分钟·第二部》
所以这个神圣的任务就交给你了。
Input
第一行一个整数n，代表数列中数的个数。
第二行n个正整数，表示初始状态下数列中的数。
第三行一个整数m，表示有m次操作。
接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。
Output
对于询问操作，每行输出一个回答。
Sample Input
10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8
Sample Output
19
7
6
HINT
1：对于100%的数据，1<=n<=100000，1<=l<=r<=n，数列中的数大于0，且不超过1e12。
2：数据不保证L<=R 若L>R，请自行交换L,R，谢谢！

Solution
同3211

Code

#include <bits/stdc++.h>
using namespace std;

#define rep(i, l, r) for (int i = (l); i <= (r); i++)
#define per(i, r, l) for (int i = (r); i >= (l); i--)
#define MS(_) memset(_, 0, sizeof(_))
#define PB push_back
#define MP make_pair
#define debug(...) fprintf(stderr, __VA_ARGS__)
typedef long long ll;
typedef pair<int, int> PII;
template<typename T> inline void read(T &x){
    x = 0; T f = 1; char ch = getchar();
    while (!isdigit(ch)) {if (ch == '-') f = -1; ch = getchar();}
    while (isdigit(ch))  {x = x * 10 + ch - '0'; ch = getchar();}
    x *= f;
}

const int N = 100000+100;
struct Node{
    int l, r, need; ll sum;
    Node *lc, *rc;
    inline void push_up(){
        sum = lc->sum + rc->sum;
        need = lc->need | rc->need;
    }   
}pool[N<<2], *tail = pool, *root = pool;
int n, m;
ll a[N];

inline void build(Node *(&rt), int l, int r){
    rt = tail++; rt->l = l; rt->r = r; rt->need = 1;
    if (l == r) { rt->sum = a[l]; return; }
    int mid = l+r >> 1;
    build(rt->lc, l, mid);
    build(rt->rc, mid+1, r);
    rt->push_up();
}
inline void modify(Node *rt, int l, int r){
    if (!rt->need) return;
    if (rt->l == rt->r) {
        rt->sum = (int)sqrt(rt->sum);
        if (rt->sum == 1) rt->need = 0;
        return;
    }
    int mid = rt->l+rt->r >> 1;
    if (l <= mid) modify(rt->lc, l, r);
    if (r > mid) modify(rt->rc, l, r);
    rt->push_up();
}
inline ll query(Node *rt, int l, int r){ ll res = 0;
    if (l <= rt->l && rt->r <= r) return rt->sum; 
    int mid = rt->l+rt->r >> 1;
    if (l <= mid) res += query(rt->lc, l, r);
    if (r > mid) res += query(rt->rc, l, r);
    return res;
}
int main(){
    read(n);
    rep(i, 1, n) read(a[i]);
    build(root, 1, n);

    read(m);
    while (m--){ int k, l, r;
        read(k); read(l); read(r); if (l > r) swap(l, r);
        if (!k) modify(root, l, r); 
        else printf("%lld\n", query(root, l, r));
    }
    return 0;   
}