POJ 3468 A Simple Problem with Integers

区间修改+区间查询

#include<iostream>
using namespace std;
const int M=100005;
struct Seb_tree
{
int l;
int r;
long long  col;
long long sum;
};
Seb_tree t[M*4+1];
void push_down(int k)
{
if (t[k].col)
{
int l=k*2;
int r=l+1;
t[l].sum+=t[k].col*(t[l].r-t[l].l+1);
t[l].col+=t[k].col;
t[r].sum+=t[k].col*(t[r].r-t[r].l+1);
t[r].col+=t[k].col;
t[k].col=0;
}
}
long long query(int l, int r, int k)
{
if (t[k].l==l && t[k].r==r)
{
   return t[k].sum;
}
push_down(k);
int mid=(t[k].r+t[k].l)/2;
if (r<=mid) query(l ,r,  k*2);
else if (l>mid) query(l, r,  k*2+1);
else return query(l, mid, k*2)+query(mid+1, r, k*2+1);
}
void updata(int l, int r, int col, int k)
{
if (t[k].l>=l && t[k].r<=r)
{
t[k].sum+=col*(r-l+1);
t[k].col+=col;
return;
}
push_down(k);
int mid=(t[k].r+t[k].l)/2;
if (r<=mid) updata(l ,r, col, k*2);
else if (l>mid) updata(l, r, col, k*2+1);
else {
updata(l, mid, col, k*2);
updata(mid+1, r, col, k*2+1);
}
t[k].sum=t[k*2].sum+t[k*2+1].sum;
}
void build(int l, int r, int k)
{
t[k].l=l;
t[k].r=r;
t[k].col=0;
if (l==r){
long long   s;
cin>>s;
t[k].sum=s;
return;
}
int mid=(l+r)/2;
build(l, mid, k*2);
build(mid+1, r, k*2+1);
t[k].sum=t[k*2] .sum+t[k*2+1].sum;
}
int main()
{
int n,m;
cin>>n>>m;
build(1, n, 1);
while (m--)
{
char  a; 
int b, c, d;
cin>>a;
if (a=='C'){
cin>>b>>c>>d;
updata(b, c, d, 1);
}
else if (a=='Q')
{
cin>>b>>c;
cout<<query(b,c,1)<<endl;
}

}
return 0;
}