1081 线段树练习 2

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 大师 Master

传送门

题目描述 Description

给你N个数，有两种操作

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

2：询问第i个数是什么？

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是1，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是2，后面跟1个整数i, 表示询问第i个位置的数是多少。

输出描述 Output Description

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

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 3

样例输出 Sample Output

5

数据范围及提示 Data Size & Hint

数据范围

1<=n<=100000

1<=q<=100000

/*

裸线段树.

单点查询+区间修改.

*/

#include<iostream>

#include<cstdio>

#define MAXN 100001

using namespace std;

struct data

{

    int l,r;

    int lc,rc;

    int sum;

    int bj;

}

tree[MAXN*4];

int aa[MAXN+10],n,m,cut,tot;

void bluid(int l,int r)

{

    int k=++cut;

    tree[k].l=l;

    tree[k].r=r;

    if(l==r)

    {

        tree[k].sum=aa[l];

        return ;

    }

    int mid=(l+r)>>1;

    tree[k].lc=cut+1;

    bluid(l,mid);

    tree[k].rc=cut+1;

    bluid(mid+1,r);

    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;

}

void updata(int k)

{

    tree[tree[k].lc].sum+=

    tree[k].bj*(tree[tree[k].lc].r-tree[tree[k].lc].l+1);

    tree[tree[k].rc].sum+=

    tree[k].bj*(tree[tree[k].rc].r-tree[tree[k].rc].l+1);

    tree[tree[k].lc].bj+=tree[k].bj;

    tree[tree[k].rc].bj+=tree[k].bj;

    tree[k].bj=0;

}

void add(int k,int l,int r,int add1)

{

    if(l<=tree[k].l&&tree[k].r<=r)

    {

        tree[k].bj+=add1;

        tree[k].sum+=add1*(tree[k].r-tree[k].l+1);

        return;

    }

    if(tree[k].bj) updata(k);

    int mid=(tree[k].l+tree[k].r)>>1;

    if(l<=mid) add(tree[k].lc,l,r,add1);

    if(r>mid) add(tree[k].rc,l,r,add1);

    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;

}

int query(int k,int l,int r)

{

    if(l<=tree[k].l&&tree[k].r<=r)

    {

        return tree[k].sum;

    }

    if(tree[k].bj) updata(k);

    int mid=(tree[k].l+tree[k].r)>>1;

    if(l<=mid) return query(tree[k].lc,l,r);

    if(r>mid) return query(tree[k].rc,l,r);

}

int main()

{

    int x,a,b,add1;

    cin>>n;

    for(int i=1;i<=n;i++)

    {

        cin>>aa[i];

    }

    bluid(1,n);

    cin>>m;

    for(int i=1;i<=m;i++)

    {

        scanf("%d",&x);

        if(x==1)

        {

            scanf("%d %d %d",&a,&b,&add1);

            add(1,a,b,add1);

        }

        else

        {

            scanf("%d",&a);

            printf("%d\n",query(1,a,a));

        }

    }

}

/*

树状数组 区间修改 单点查询.

比线段树不知道要快到那里去.

和区间修改维护的东西不同.

差分化区间修改为单点修改.

例:[2,5]+2  只需处+2  6处-2即可.

*/

#include<cstdio>

#include<iostream>

#define MAXN 100001

int s[MAXN],n,x,y,m,z;

int read(){

    int x=0,f=1;char ch=getchar();

    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}

    while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=getchar();

    return x*f;

}

int lowbit(int t){

    return t&-t;

}

void add(int t,int x){

    while(t<=n){

        s[t]+=x;

        t+=lowbit(t);

    }

}

int query(int x){

    int tot=0;

    while(x){

        tot+=s[x];

        x-=lowbit(x);

    }

    return tot;

}

int main(){

    n=read();

    for(int i=1;i<=n;i++){

        x=read();

        add(i,x);add(i+1,-x);

}

    m=read();

    for(int i=1;i<=m;i++){

        z=read();

        if(z==1){

            x=read();y=read();z=read();add(x,z);add(y+1,-z);

        }

        else{

            x=read();printf("%d\n",query(x));

        }

    }

    return 0;

}