poj3468A Simple Problem with Integers

题目链接：http://poj.org/problem?id=3468

区间更新，区间求和。

 1 #include<cstdio>
 2 #include<algorithm>
 3 using namespace std;
 4 #define ll long long
 5 #define lson l,m,rt<<1
 6 #define rson m+1,r,rt<<1|1
 7 const int maxn=100010;
 8 ll add[maxn<<2];
 9 ll sum[maxn<<2];
10
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)
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
28 void build(int l,int r,int rt)
29 {
30             add[rt]=0;
31     if(l==r) {
32         scanf("%lld",&sum[rt]);
33         return;
34     }
35     int m=(l+r)>>1;
36     build(lson);
37     build(rson);
38     pushup(rt);
39 }
40
41 void update(int L,int R,int c,int l,int r,int rt)
42 {
43     if(L<=l&&r<=R){
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(R>m) update(L,R,c,rson);
52     pushup(rt);
53 }
54
55 ll query(int L,int R,int l,int r,int rt)
56 {
57     if(L<=l&&r<=R)
58     {
59         return sum[rt];
60     }
61     pushdown(rt,r-l+1);
62     ll ans=0;
63     int m=(l+r)>>1;
64     if(L<=m) ans+=query(L,R,lson);
65     if(R>m) ans+=query(L,R,rson);
66     return ans;
67 }
68
69 int main()
70 {
71     int n,m;
72     while(scanf("%d%d",&n,&m)!=EOF)
73     {
74
75         build(1,n,1);
76         while(m--)
77         {
78             int a,b,c;
79             char s[2];
80             scanf("%s",s);
81             if(s[0]==‘Q‘)
82             {
83                 scanf("%d%d",&a,&b);
84                 printf("%lld\n",query(a,b,1,n,1));
85             }
86             else {
87                 scanf("%d%d%d",&a,&b,&c);
88                 update(a,b,c,1,n,1);
89             }
90
91         }
92     }
93     return 0;
94 }