BZOJ 3038: 上帝造题的七分钟2

 

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

 

 

——我是愉快的分隔符——

这道题就是一个加了Lazy Tag的线段树，写的比较烦，但是应该很好想。

代码：

#include<cstdio>

#include<cmath>

#include<iostream>

using namespace std;

const int maxn=100000;

long long num[maxn+5];

long long value[maxn+5];

int next[maxn+5];

int n,m;

inline int lowbit(int x){return x&-x;}

inline void swap(int &x,int &y){x=x+y;y=x-y;x=x-y;}

inline void add(int x,long long val){

       for (int i=x;i<=maxn;i+=lowbit(i))

           num[i]+=val;

}

inline long long get(int x){

       long long ans=0;

       for (int i=x;i>0;i-=lowbit(i))

           ans+=num[i];

       return ans;

}          

inline void initNext(){

       for (int i=1;i<=n;i++) next[i]=i;

}  

inline int getNext(int x){

       return next[x]=next[x]==x?x:getNext(next[x]);

}

inline void solve(int l,int r){

       if (l>r) swap(l,r);

       for (int i=l;i<=r;i++){

           if (next[i]!=i){

              i=next[i]=getNext(i);

              if (i>r) return;

           }

           if (value[i]==1) continue;

           long long temp=value[i];

           value[i]=(long long)sqrt(value[i]);

           add(i,(long long)value[i]-temp);

           if (value[i]==1) next[i]=getNext(i+1);

       }   

}

int main(){



    scanf("%d",&n);

    for (int i=1;i<=n;i++){

        scanf("%lld",&value[i]);

        add(i,value[i]);

    }

    initNext();

    scanf("%d",&m);

    int k,l,r;

    for (int i=1;i<=m;i++){

        int k,l,r;

        scanf("%d%d%d",&k,&l,&r);

        if (k==0){

           solve(l,r);

        }else{

              if (l>r) swap(l,r);

              printf("%lld\n",(long long)get(r)-get(l-1));

        }

    }



    return 0;

}