bzoj 3211 花神游历各国

 1 #include<cstdio>
 2 #include<cmath>
 3 #define MAXN 400010
 4 #define LL long long
 5 LL sumv[MAXN],flag[MAXN];
 6 int n,m,t1,yl,yr;
 7 void build(int o,int L,int R)
 8 {
 9     int lc=o<<1,rc=(o<<1)+1,M=(L+R)/2;
10     if(L==R)
11     {
12         scanf("%lld",&sumv[o]);
13         return;
14     }
15     build(lc,L,M);
16     build(rc,M+1,R);
17     sumv[o]=sumv[lc]+sumv[rc];
18     flag[o]=flag[lc]&flag[rc];
19 }
20 void update(int o,int L,int R)
21 {
22     if(flag[o])return;
23     int lc=o<<1,rc=(o<<1)+1,M=(L+R)/2;
24     if(L==R)
25     {
26         sumv[o]=sqrt(sumv[o]);
27         if(!sumv[o]||sumv[o]==1)flag[o]=true;
28         return;
29     }
30     if(yl<=M)update(lc,L,M);
31     if(yr>M)update(rc,M+1,R);
32     sumv[o]=sumv[lc]+sumv[rc];
33     flag[o]=flag[lc]&flag[rc];
34 }
35 LL query(int o,int L,int R)
36 {
37     int lc=o<<1,rc=(o<<1)+1,M=(L+R)>>1;
38     if(yl<=L&&yr>=R)return sumv[o];
39     LL ans=0;
40     if(yl<=M)ans=query(lc,L,M);
41     if(yr>M)ans+=query(rc,M+1,R);
42     sumv[o]=sumv[lc]+sumv[rc];
43     return ans;
44 }
45 int main()
46 {
47     scanf("%d",&n);
48     build(1,1,n);
49     scanf("%d",&m);
50     while(m--)
51     {
52         scanf("%d%d%d",&t1,&yl,&yr);
53         if(t1==2)update(1,1,n);
54         else printf("%lld\n",query(1,1,n));
55     }
56     return 0;
57 }