线段树练习2

最新推荐文章于 2021-12-14 22:50:31 发布

原创最新推荐文章于 2021-12-14 22:50:31 发布 · 314 阅读

0 ·

CC 4.0 BY-SA版权

树专栏收录该内容

5 篇文章

订阅专栏

#include<iostream>
#include<cstdio>
#include<cstdlib>//wym's second xianduanshu
#define maxn 7000000
#define ls (num<<1)
#define rs (num<<1|1)
using namespace std;
int a[maxn],tree[maxn],size[maxn],lazy[maxn];
void build (int num,int l,int r)
{
    size[num]=r-l+1;
    if (l==r) 
    {
        tree[num]=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    build (ls,l,mid);
    build (rs,mid+1,r);
    tree[num]=tree[ls]+tree[rs];
}
void down (int num)
{
    tree[ls]+=lazy[num]*size[ls];
    tree[rs]+=lazy[num]*size[rs];
    lazy[ls]+=lazy[num],lazy[rs]+=lazy[num];
    lazy[num]=0;
}
void update(int num,int l,int r,int x,int y,int z)
{
    if (l!=r) down(num);
    if (l==x&&r==y) 
    {
        tree[num]+=z*size[num];
        lazy[num]+=z;
        return ;
    }
    int mid=(l+r)>>1;
    if (y<=mid) update(ls,l,mid,x,y,z);
    else if (x>mid)  update(rs,mid+1,r,x,y,z);
    else update(ls,l,mid,x,mid,z),update(rs,mid+1,r,mid+1,y,z);
    tree[num]=tree[ls]+tree[rs];
}
int query(int num,int l,int r,int x,int y)
{
    if (l!=r) down(num);
    if (x<=l&&r<=y) return tree[num];
    int mid=(l+r)>>1;
    if (y<=mid) return query(ls,l,mid,x,y);
    else if (x>mid) return query(rs,mid+1,r,x,y);
    else return query(ls,l,mid,x,mid)+query(rs,mid+1,r,mid+1,y);
}
/*int query(int num,int l,int r,int x,int y)
{
    if (l!=r) down(num);
    if (x==l) return tree[num];
    int mid=(l+r)>>1,res=0;
    if (x<=mid) res+=query(ls,l,mid,x,y);
    if (y>mid)  res+=query(rs,mid+1,r,x,y);
    return res;
}*/
int main()
{
    int n,m;
    scanf ("%d",&n);
    for (int i=1;i<=n;++i)
        scanf ("%d",&a[i]);
    build (1,1,n);
    scanf ("%d",&m);
    for (int i=1;i<=m;++i)
    {
        int b,x,y,z;
        scanf ("%d",&b);
        if (b==1)
        {
            scanf ("%d%d%d",&x,&y,&z);
            update(1,1,n,x,y,z);
        }
        else  
        {
            scanf ("%d",&x);
            printf("%d\n",query(1,1,n,x,x));
        }
    }
    return 0;
}