bzoj3038: 上帝造题的七分钟2 线段树

bzoj3038: 上帝造题的七分钟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，谢谢！

分析

这题的难点就是注意到这个平法操作操作6次之后就无效了。因为1e12的数操作六次之后就是1了。
所以我们只要用线段树维护所有数的和，对于大于1的数暴力修改，对于小于等于1的数区间求和就好了。

代码

/**************************************************************
    Problem: 3038
    User: 2014lvzelong
    Language: C++
    Result: Accepted
    Time:964 ms
    Memory:6028 kb
****************************************************************/

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<cmath>
using namespace std;
const int N = 110000;
long long read() {
    char ch = getchar(); long long x = 0;
    while(ch < '0' || ch > '9') ch = getchar();
    for(;ch >= '0' && ch <= '9'; ch = getchar()) x = (x << 1) + (x << 3) - '0' + ch;
    return x;
}
long long sum[N << 2], a[N];
bool tag[N << 2];
void Update(int p) {
    sum[p] = sum[p << 1] + sum[p << 1 | 1];
    tag[p] = tag[p << 1] && tag[p << 1 | 1];
}
void Build(int p, int L, int R) {
    if(L == R) {sum[p] = a[L]; tag[p] = a[L] <= 1; return;}
    int mid = L + R >> 1;
    Build(p << 1, L, mid);
    Build(p << 1 | 1, mid + 1, R);
    Update(p);
}
void Change(int p, int L, int R, int st, int ed) {
    if(tag[p]) return ;
    if(L == R) {
        sum[p] = sqrt(sum[p]);
        tag[p] = sum[p] <= 1;
        return;
    }
    int mid = L + R >> 1;
    if(st <= mid) Change(p << 1, L, mid, st, min(mid, ed));
    if(ed > mid) Change(p << 1 | 1, mid + 1, R, max(st, mid + 1), ed);
    Update(p);
}
long long Query(int p, int L, int R, int st, int ed) {
    if(L == st && ed == R) return sum[p];
    int mid = L + R >> 1; long long ans = 0;
    if(st <= mid) ans += Query(p << 1, L, mid, st, min(mid, ed));
    if(ed > mid) ans += Query(p << 1 | 1, mid + 1, R, max(st, mid + 1), ed);
    return ans;
}

int main() {
    int n = read();
    for(int i = 1;i <= n; ++i) a[i] = read();
    Build(1, 1, n); int m = read();
    while(m--) {
        int opt = read(), L = read(), R = read();
        if(L > R) swap(L, R);
        if(!opt) Change(1, 1, n, L, R);
        else printf("%lld\n", Query(1, 1, n, L, R));
    }
    return 0; 
}