1082 線段樹練習 3

給你N個數，有兩種操作：

1：給區間[a,b]的所有數增加X

2：詢問區間[a,b]的數的和。

第一行一個正整數n，接下來n行n個整數，

再接下來一個正整數Q，每行表示操作的個數，

如果第一個數是1，後接3個正整數，

表示在區間[a,b]內每個數增加X,如果是2，

表示操作2詢問區間[a,b]的和是多少。

pascal選手請不要使用readln讀入

對於每個詢問輸出一行一個答案

3

1

2

3

2

1 2 3 2

2 2 3

9

數據範圍

1<=n<=200000

1<=q<=200000

線段樹：區間增減 和 區間查詢。

 1
 
 #include<cstdio>
 2 #define lson l,m,rt << 1 
 3 #define rson m+1,r,rt<<1|1 
 4 #define ll long long 
 5 
 6 const int N = 200010;
 7 
 8 ll sum[N<<2];
 9 ll add[N<<2];
10 int n,q;
11 
12 void pushup(int rt)
13 {
14     sum[rt] = sum[rt<<1] + sum[rt<<1
 
|1];
15 }
16 void pushdown(int rt,int m)  //m長度，rt線段 
17 {
18     if(add[rt])
19     {
20         add[rt<<1] += add[rt] ;
21         add[rt<<1|1] += add[rt];
22         sum[rt<<1] += add[rt] * (m - (m>>1));
23         sum[rt<<1|1] += add[rt] * (m>>1);
24         add[rt] = 0;
25     }
26 }
27 void build(int l,int r,int rt)
28 {
29     add[rt] = 0;
30     if(l==r)
31     {
32         scanf("%d",&sum[rt]);
33         return ;
34     }
35     int m = (l+r)>>1;
36     build(lson);
37     build(rson);
38     pushup(rt);    
39 }
40 void update(int L,int R,int c,int l,int r,int rt)
41 {
42     if(L<=l && r<=R)
43     {
44         add[rt] += c;
45         sum[rt] += (ll)c * (r-l+1);
46         return ;
47     }
48     pushdown(rt,r-l+1);
49     int m = (l+r)>>1;
50     if(L<=m) update(L,R,c,lson);
51     if(m<R) update(L,R,c,rson);
52     pushup(rt);
53 }
54 ll query(int L,int R,int l,int r,int rt)
55 {
56     if(L<=l && r<=R)
57     {
58         return sum[rt];
59     }
60     pushdown(rt,r-l+1);
61     ll ret=0;
62     int m = (l+r)>>1;
63     if(L<=m) ret += query(L,R,lson);
64     if(m<R) ret += query(L,R,rson);
65     return ret;
66 }
67 int main()
68 {
69     scanf("%d",&n);
70     build(1,n,1);
71     scanf("%d",&q);
72     while(q--)
73     {
74         int a,b,c,x;
75         scanf("%d",&x);
76         if(x==1)
77         {
78             scanf("%d%d%d",&a,&b,&c);
79             update(a,b,c,1,n,1);
80         }
81         else
82         {
83             scanf("%d%d",&a,&b);
84             printf("%lld\n",query(a,b,1,n,1));
85         }
86     }
87     return 0;
88 }

1082 線段樹練習 3