bzoj 3038 上帝造题的七分钟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

题解

最近狂刷线段树……跟着某大佬的后面刷Ta刷过的题……蹭之题解。本题之妙在于开方能使一个区间快速归1，且其将永无翻身之时，所以不必再向下更新。只需维护一个max，在modify时特判该区间max是否是1，如果是1，那这个区间都是1，没有更新的必要，可以直接略过了。

代码

#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 1000010;
long long a[N];
struct Node {
    int l, r;
    long long sum, maxx;
} t[N];
int n, m, q, l, r, root;

void build(int root, int L, int R) {
    t[root].l = L, t[root].r = R;
    if(L == R) {
        t[root].sum = a[L];
        t[root].maxx = a[L];
        return ;
    }
    int mid = (L + R) >> 1;
    build(root << 1, L, mid);
    build(root << 1 | 1, mid + 1, R);
    t[root].sum = t[root << 1].sum + t[root << 1 | 1].sum;
    t[root].maxx = max(t[root << 1].maxx, t[root << 1 | 1].maxx);
    return ;
}

void modify(int root, int L, int R) {
    int l = t[root].l, r = t[root].r;
    if(l == r) {
        t[root].sum  = sqrt(t[root].sum);
        t[root].maxx = sqrt(t[root].maxx);
        return ;
    }
    int mid = (l + r) >> 1;
    if(L <= mid && t[root<<1].maxx > 1)
        modify(root << 1, L, R);
    if(R > mid && t[root<<1 | 1].maxx > 1)
        modify(root << 1 | 1, L, R);
    t[root].sum = t[root << 1].sum + t[root << 1 | 1].sum;
    t[root].maxx = max(t[root << 1].maxx, t[root << 1 | 1].maxx);
    return ;
}

long long query(int root, int L, int R) {
    int l = t[root].l, r = t[root].r;
    if(L <= l && r <= R)
        return t[root].sum;
    long long rt = 0;
    int mid = (l + r) >> 1;
    if(L <= mid)
        rt += query(root << 1, L, R);
    if(R > mid)
        rt += query(root << 1 | 1, L, R);
    return rt;
}

int main() {
    freopen("tt.in", "r", stdin);
    freopen("tt.out", "w", stdout);
    scanf("%d", &n);
    for(register int i = 1; i <= n; i ++)
        scanf("%lld", &a[i]);
    scanf("%d", &m);
    build(root = 1, 1, n);
    while(m --) {
        scanf("%d %d %d", &q, &l, &r);
        if(l > r) swap(l, r);
        if(! q) modify(1, l, r);
        else printf("%lld\n", query(1, l, r));
    }
}