POJ A Simple Problem with Integers | 线段树基础练习

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 typedef long long ll;
 #define N 100010
 using namespace std;
 struct node
 {
     ll l,r,lz,sum;
 }t[*N];
 ll read()
 {
     ll ret=,neg=;
     char j=getchar();
     for (;j>'' || j<'';j=getchar())
     if (j=='-') neg=-;
     for (;j<='' && j>='';j=getchar())
     ret=ret*+j-'';
     return ret*neg;
 }
 ll n,q,a[N],l,r,k;
 void pushup(ll p)
 {
     t[p].sum=t[*p].sum+t[*p+].sum;
 }
 void pushdown(ll p)
 {
     if (t[p].l==t[p].r || t[p].lz==) return;
     ll tmp=t[p].lz,mid=(t[p].r+t[p].l)>>;
     t[*p].sum+=(mid-t[p].l+)*tmp;
     t[*p+].sum+=(t[p].r-mid)*tmp;
     t[*p].lz+=tmp;
     t[*p+].lz+=tmp;
     t[p].lz=;
 }
 void build(ll p,ll l,ll r)
 {
     t[p].l=l,t[p].r=r;
     if (l!=r)
     {
     ll mid=l+r>>;
     build(*p,l,mid);
     build(*p+,mid+,r);
     pushup(p);
     }
     else
     t[p].sum=a[l];
 }
 void modify(ll p,ll l,ll r,ll k)
 {
     ll L=t[p].l,R=t[p].r,mid=(L+R)>>;
     if (l==L && r==R)
     {
     t[p].sum+=(r-l+)*k;
     t[p].lz+=k;
     return ;
     }
     pushdown(p);
     if (r<=mid)
     modify(*p,l,r,k);
     else if (l>mid)
     modify(*p+,l,r,k);
     else modify(*p,l,mid,k),modify(*p+,mid+,r,k);
     pushup(p);
 }
 ll query(ll p,ll l,ll r)
 {
     ll L=t[p].l,R=t[p].r,mid=(L+R)>>;
     if (l==L && r==R)
     return t[p].sum;
     pushdown(p);
     if (r<=mid)
     return query(*p,l,r);
     if (l>mid) return query(*p+,l,r);
     return query(*p,l,mid)+query(*p+,mid+,r);
 }
 char op[N];
 int main()
 {
     n=read(),q=read();
     for (int i=;i<=n;i++)
     a[i]=read();
     build(,,n);
     while (q--)
     {
     scanf("%s",op);
     if (op[]=='Q')
         l=read(),r=read(),printf("%lld\n",query(,l,r));
     else l=read(),r=read(),k=read(),modify(,l,r,k);
     }
     return ;
 }