【线段树/区间开平方】BZOJ3211-花神游历各国

【题目大意】

给出一些数，有两种操作。(1)将区间内每一个数开方(2)查询每一段区间的和

【思路】

普通的线段树保留修改+开方优化。可以知道当一个数为0或1时，无论开方几次，答案仍然相同。所以设置flag=1。如果一个节点的左右孩子flag均为1，那么它的flag也是1。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 #define lson l,m,rt<<1
 #define rson m+1,r,rt<<1|1
 #define LL long long
 using namespace std;
 const int MAXN=+;
 int n,m;
 LL sum[MAXN<<];
 int flag[MAXN<<];

 void pushup(int rt)
 {
     sum[rt]=sum[rt<<]+sum[rt<<|];
     flag[rt]=flag[rt<<]&flag[rt<<|];
 }

 void build(int l,int r,int rt)
 {
     flag[rt]=;
     if (l==r)
     {
         scanf("%lld",&sum[rt]);
         return;
     }
     int m=(l+r)>>;
     build(lson);
     build(rson);
     pushup(rt);
 }

 void update(int L,int R,int l,int r,int rt)
 {
     if (flag[rt]) return;
     if (l==r)
     {
         sum[rt]=(LL)sqrt(sum[rt]);
         if (sum[rt]== || sum[rt]==) flag[rt]=;
         return;
     }
     int m=(l+r)>>;
     if (L<=m) update(L,R,lson);
     if (R>m) update(L,R,rson);
     pushup(rt);
 }

 LL query(int L,int R,int l,int r,int rt)
 {
     if (L<=l && r<=R) return sum[rt];
     int m=(l+r)>>;
     LL ret=;
     if (L<=m) ret+=query(L,R,lson);
     if (R>m) ret+=query(L,R,rson);
     return ret;
 }

 int main()
 {
     scanf("%d",&n);
     build(,n,);
     scanf("%d",&m);
     for (int i=;i<m;i++)
     {
         int x,l,r;
         scanf("%d%d%d",&x,&l,&r);
         if (x==) printf("%lld\n",query(l,r,,n,));
             else update(l,r,,n,);
     }
     return ;
 }