POJ 3468 A Simple Problem with Integers 线段树

Description

给出了一个序列，你需要处理如下两种询问。

"C a b c"表示给[a, b]区间中的值全部增加c (-10000 ≤ c ≤ 10000)。

"Q a b" 询问[a, b]区间中所有值的和。

Input

第一行包含两个整数N, Q。1 ≤ N,Q ≤ 100000.

第二行包含n个整数，表示初始的序列A (-1000000000 ≤ Ai ≤ 1000000000)。

接下来Q行询问，格式如题目描述。

Output

对于每一个Q开头的询问，你需要输出相应的答案，每个答案一行。

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

线段树题，注意这题数据比较大要用long long定义变量。

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84

#include<cstring> #include<cstdio> using namespace std; struct p {     long long x,y,z,v; };p tree[1000000]; long long a[100010],sum; long long build(long long l,long long r,long long p) {     tree[p].x=l;     tree[p].y=r;     tree[p].v =0;     if (l==r) return tree[p].z=a[l];     long long m=(l+r)/2;     long long a=build(l,m,2*p);     long long b=build(m+1,r,2*p+1);     return tree[p].z=a+b; } long long un(long long l,long long r,long long z,long long p) {     if (tree[p].x==l&&tree[p].y==r) {tree[p].v+=z; tree[p].z+=z*(r-l+1);return tree[p].z;}     if (tree[p].v)     {         tree[2*p].v+=tree[p].v;         tree[2*p].z+=tree[p].v*(tree[2*p].y-tree[2*p].x+1);         tree[2*p+1].v+=tree[p].v;         tree[2*p+1].z+=tree[p].v*(tree[2*p+1].y-tree[2*p+1].x+1);         tree[p].v=0;     }     long long m=(tree[p].x+tree[p].y)/2;     if (l>m) return tree[p].z=un(l,r,z,2*p+1)+tree[2*p].z;     if (m>=r) return tree[p].z=un(l,r,z,2*p)+tree[2*p+1].z;     return tree[p].z=un(l,m,z,2*p)+un(m+1,r,z,2*p+1); } void find(long long l,long long r,long long p) {     if (tree[p].x==l&&tree[p].y==r) {sum+=tree[p].z;return ;}     if (tree[p].v)     {         tree[2*p].v+=tree[p].v;         tree[2*p].z+=tree[p].v*(tree[2*p].y-tree[2*p].x+1);         tree[2*p+1].v+=tree[p].v;         tree[2*p+1].z+=tree[p].v*(tree[2*p+1].y-tree[2*p+1].x+1);         tree[p].v=0;     }     long long m=(tree[p].x+tree[p].y)/2;     if (l>m) find(l,r,2*p+1);     else if (m>=r) find(l,r,2*p);     else     {         find(l,m,2*p);         find(m+1,r,2*p+1);     }     return ; } int main() {     long long n,q,i,x,y,z;     char s;     while (~scanf("%lld%lld",&n,&q))     {         for (i=1;i<=n;i++) scanf("%lld",&a[i]);         build(1,n,1);         while (q--)         {             //getchar();             scanf(" %c",&s);             if (s=='C')             {                 scanf("%lld%lld%lld",&x,&y,&z);                 un(x,y,z,1);             }             else             {                 scanf("%lld%lld",&x,&y);                 sum=0;                 find(x,y,1);                 printf("%lld\n",sum);             }         }     }     return 0; }