bzoj 3212 Pku3468 A Simple Problem with Integers

3212: Pku3468 A Simple Problem with Integers

Time Limit: 1 Sec  Memory Limit: 128 MB

Description

You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval.

Input

The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000. 
The second line contains N numbers, the initial values of A1, A2, ... , AN. -1000000000 ≤ Ai ≤ 1000000000. 
Each of the next Q lines represents an operation. 
"C a b c" means adding c to each of Aa, Aa+1, ... , Ab. -10000 ≤ c ≤ 10000. 
"Q a b" means querying the sum of Aa, Aa+1, ... , Ab.

Output

You need to answer all Q commands in order. One answer in a line.

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

4
55
9
15

HINT

The sums may exceed the range of 32-bit integers.

Source

 

Tips：

　　非常裸的线段树的区间操作；

 

Code：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
int n,m,a[];
long long tree[],mark[];
char s[];

void build(int l,int r,int v){
    if(l==r){
        tree[v]=a[l];
        return;
    }
    int mid=(l+r) >> ;
    build(l,mid,v<<);
    build(mid+,r,(v<<)+);
    tree[v]=tree[v<<]+tree[(v<<)+];
}

void add(int l,int r,int x,int y,int z,int v){
    if(l==x&&r==y){
        mark[v]+=z;
        return;
    }
    int mid=(l+r)>>;
    mark[v<<]+=mark[v];
    mark[(v<<)+]+=mark[v];
    mark[v]=;
    if(mid>=y)  add(l,mid,x,y,z,v<<);
    if(mid<x)  add(mid+,r,x,y,z,(v<<)+);
    if(mid>=x&&mid<y){
        add(l,mid,x,mid,z,v<<);
        add(mid+,r,mid+,y,z,(v<<)+);
    }
    tree[v]=tree[v<<]+mark[v<<]*(mid-l+)+tree[(v<<)+]+mark[(v<<)+]*(r-mid);
}

long long query(int l,int r,int x,int y,int v){
    long long ans=;
    if(l==x&&r==y){
        ans=tree[v]+mark[v]*(r-l+);
        return ans;
    }
    mark[v<<]+=mark[v];
    mark[(v<<)+]+=mark[v];
    mark[v]=;
    int mid=(l+r)>>;
    if(mid>=y)  ans=query(l,mid,x,y,v<<);
    if(mid<x)  ans=query(mid+,r,x,y,(v<<)+);
    if(mid>=x&&mid<y)  ans=query(l,mid,x,mid,v<<)+query(mid+,r,mid+,y,(v<<)+);
    tree[v]=tree[v<<]+mark[v<<]*(mid-l+)+tree[(v<<)+]+mark[(v<<)+]*(r-mid);
    return ans;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
    }
    build(,n,);
    for(int i=;i<=m;i++){
        scanf("%s",s);
        int x,y,z;
        if(s[]=='C'){
            scanf("%d%d%d",&x,&y,&z);
            if(x>y) swap(x,y);
            add(,n,x,y,z,);
        }else{
            scanf("%d%d",&x,&y);
            if(x>y) swap(x,y);
            printf("%lld\n",query(,n,x,y,));
        }
    }
}