BZOJ3038上帝造题的七分钟2

3038: 上帝造题的七分钟2
Time Limit: 3 Sec Memory Limit: 128 MB
Submit: 860 Solved: 387
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，谢谢！
Source
Poetize4
本来以为是和花神游历各国一样的水题。。
用线段树编了一下，居然卡了我半个小时，我是不是该退役了233
直接编的话会TLE，所以需要加一个小小的优化。。
也就是说当数列值为0或1时，便不需要开根修改。。
于是加了这个小优化后便AC了。。

#include
#include
#include
using namespace std;
long long a[100001],ans;
int n,m;
struct edge
{
    long long value;
    bool flag;
};
edge node[400001];

long long read()
{
    long long w=0,c=1;
    char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-')
          c=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      {
        w=w*10+ch-'0';
        ch=getchar();
      }
    return w*c;
}

void update(int s)
{
    node[s].value=node[s*2].value+node[s*2+1].value;
}

void build(int s,int l,int r)
{
    if (l==r)
      {
        node[s].value=a[l];
        if (a[l]==1 || a[l]==0)
          node[s].flag=true;
        return;
      }
    build(s*2,l,(l+r)/2);
    build(s*2+1,(l+r)/2+1,r);
    update(s);
    node[s].flag=node[s*2].flag&node[s*2+1].flag;
}

void insert(int s,int l,int r,int x,int y)
{
    if (node[s].flag)
      return;
    int mid=(l+r)/2;
    if (l==r)
      {
        node[s].value=(long long)sqrt(node[s].value);
        if (node[s].value==1 || node[s].value==0)
          node[s].flag=true;
        return;
      }
    if (x<=mid)
      insert(s*2,l,mid,x,y);
    if (y>mid)
      insert(s*2+1,mid+1,r,x,y);
    update(s);
    node[s].flag=node[s*2].flag&node[s*2+1].flag;
}

void query(int s,int l,int r,int x,int y)
{
    int mid=(l+r)/2;
    if (x<=l && y>=r)
      {
        ans+=node[s].value;
        return;
      }
    if (x<=mid)
      query(s*2,l,mid,x,y);
    if (y>mid)
      query(s*2+1,mid+1,r,x,y);
}

int main()
{
    int i,p,x,y;
    scanf("%d",&n);
    for (i=1;i<=n;i++)
      a[i]=read();
    scanf("%d",&m);
    build(1,1,n);
    for (i=1;i<=m;i++)
      {
        scanf("%d %d %d",&p,&x,&y);
        if (x>y)
          swap(x,y);
        if (p==0)
          insert(1,1,n,x,y);
        if (p==1)
          {
            ans=0;
            query(1,1,n,x,y);
            printf("%lld\n",ans);
          }
      }
    return 0;
}