线段树练习3

 1 #include<iostream>
 2 #include<cstdio>
 3 #define ll long long
 4 
 5 using namespace std;
 6 const int N=200002;
 7 
 8 struct node{
 9     long long l,r,w,f;
10 }T[N*4];
11 
12 long long z,x,y,yj;
13 long long ans;
14 long long n,m;
15 
16 void down(ll jd)
17 {
18     T[jd<<1].w+=T[jd].f*(T[jd<<1].r-T[jd<<1].l+1);
19     T[jd<<1|1].w+=T[jd].f*(T[jd<<1|1].r-T[jd<<1|1].l+1);
20     T[jd<<1].f+=T[jd].f;
21     T[jd<<1|1].f+=T[jd].f;
22     T[jd].f=0;
23 }
24 
25 void build(ll l,ll r,ll jd)
26 {
27     T[jd].l=l;
28     T[jd].r=r;
29     if(T[jd].l==T[jd].r)
30     {
31         scanf("%lld",&T[jd].w);
32         return ;
33     }
34     int mid=(l+r)>>1;
35     build(l,mid,jd<<1);
36     build(mid+1,r,jd<<1|1);
37     T[jd].w=T[jd<<1].w+T[jd<<1|1].w;
38 }
39 
40 void qj_gai(ll jd)
41 {
42     if(x<=T[jd].l&&T[jd].r<=y)
43     {
44         T[jd].w+=(T[jd].r-T[jd].l+1)*yj;
45         T[jd].f+=yj;
46         return ;
47     }
48     if(T[jd].f)down(jd);
49     int mid=(T[jd].l+T[jd].r)>>1;
50     if(x<=mid)qj_gai(jd<<1);
51     if(y>mid)qj_gai(jd<<1|1);
52     T[jd].w=T[jd<<1].w+T[jd<<1|1].w;
53 }
54 
55 void qj_ask(ll jd)
56 {
57     if(x<=T[jd].l&&T[jd].r<=y)
58     {
59         ans+=T[jd].w;
60         return ;
61     }
62     if(T[jd].f)down(jd);
63     int mid=(T[jd].l+T[jd].r)>>1;
64     if(x<=mid)qj_ask(jd<<1);
65     if(y>mid)qj_ask(jd<<1|1);
66 }
67 
68 int main()
69 {
70     scanf("%lld",&n);
71     build(1,n,1);
72     scanf("%lld",&m);
73     for(int i=1;i<=m;i++)
74     {
75         scanf("%lld",&z);
76         if(z==1)
77         {
78             scanf("%lld%lld%lld",&x,&y,&yj);
79             qj_gai(1);
80         }
81         if(z==2)
82         {
83             scanf("%lld%lld",&x,&y);
84             ans=0;
85             qj_ask(1);
86             printf("%lld\n",ans);
87         }
88      }
89      return 0; 
90 }