1082 线段树练习 3

1082 线段树练习 3 

 时间限制: 3 s

 空间限制: 128000 KB

 题目等级 : 大师 Master 

 

题目描述 Description

给你N个数，有两种操作：

1：给区间[a,b]的所有数增加X

2：询问区间[a,b]的数的和。

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，

再接下来一个正整数Q，每行表示操作的个数，

如果第一个数是1，后接3个正整数，

表示在区间[a,b]内每个数增加X,如果是2，

表示操作2询问区间[a,b]的和是多少。

pascal选手请不要使用readln读入

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 2 3

样例输出 Sample Output

9

数据范围及提示 Data Size & Hint

数据范围

1<=n<=200000

1<=q<=200000

线段树：区间增减 和 区间查询。

#include<cstdio>
#define lson l,m,rt << 1 
#define rson m+1,r,rt<<1|1 
#define ll long long 

const int N = 200010;

ll sum[N<<2];
ll add[N<<2];
int n,q;

void pushup(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void pushdown(int rt,int m)  //m长度，rt线段 
{
    if(add[rt])
    {
        add[rt<<1] += add[rt] ;
        add[rt<<1|1] += add[rt];
        sum[rt<<1] += add[rt] * (m - (m>>1));
        sum[rt<<1|1] += add[rt] * (m>>1);
        add[rt] = 0;
    }
}
void build(int l,int r,int rt)
{
    add[rt] = 0;
    if(l==r)
    {
        scanf("%d",&sum[rt]);
        return ;
    }
    int m = (l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);    
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l && r<=R)
    {
        add[rt] += c;
        sum[rt] += (ll)c * (r-l+1);
        return ;
    }
    pushdown(rt,r-l+1);
    int m = (l+r)>>1;
    if(L<=m) update(L,R,c,lson);
    if(m<R) update(L,R,c,rson);
    pushup(rt);
}
ll query(int L,int R,int l,int r,int rt)
{
    if(L<=l && r<=R)
    {
        return sum[rt];
    }
    pushdown(rt,r-l+1);
    ll ret=0;
    int m = (l+r)>>1;
    if(L<=m) ret += query(L,R,lson);
    if(m<R) ret += query(L,R,rson);
    return ret;
}
int main()
{
    scanf("%d",&n);
    build(1,n,1);
    scanf("%d",&q);
    while(q--)
    {
        int a,b,c,x;
        scanf("%d",&x);
        if(x==1)
        {
            scanf("%d%d%d",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
        else
        {
            scanf("%d%d",&a,&b);
            printf("%lld
",query(a,b,1,n,1));
        }
    }
    return 0;
}

2018-07-08

涨姿势，树状数组区间求和，区间修改。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;

inline int read() {
    int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
    for (;isdigit(ch);ch=getchar())x=x*10+ch-'0';return x*f;
}

const int N = 200010;
int a[N],b[N];

struct BIT{
    int n;LL c1[N],c2[N];
    void update(int p,int v) {
        for (int i=p; i<=n; i+=(i&(-i))) c1[i] += v,c2[i] += p*v;
    }
    LL query(int p) {
        LL ans = 0;
        for (int i=p; i; i-=(i&(-i))) ans += (p+1)*c1[i]-c2[i];
        return ans;
    }
}bit;

int main() {
    int n = read(),last = 0;
    bit.n = n;
    for (int i=1; i<=n; ++i) {
        int tmp = read();
        a[i] = tmp - last;
        last = tmp;
    }
    for (int i=1; i<=n; ++i) 
        bit.update(i,a[i]);
        
    int m = read();
    while (m--) {
        int opt = read();
        if (opt == 1) {
            int l = read(),r = read(),v = read();
            bit.update(l,v);
            if (r < n) bit.update(r+1,-v);
        }
        else {
            int l = read(),r = read();
            printf("%lld
",bit.query(r)-bit.query(l-1));
        }
    }
    return 0;
}