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定义变量。

#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;
}