区间修改,单点查询(树状数组)

#include"stdio.h"
#include"iostream"
#include"stdlib.h"
#include"math.h"
using namespace std;
#define MAXN 5000006
int tree[MAXN],a[MAXN];
int n,m,q,w,e,r;
int lowbit(int x)
{return x&(-x);}
int Sum(int x)
{
    int s=0;
    while(x>0)
    {
        s+=tree[x];
        x-=lowbit(x);
    }
    return s;
}
void Add(int x,int k)
{
    while(x<=n)
    {
        tree[x]+=k;
        x+=lowbit(x);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int last=0;
    for(int i=1;i<=n;i++)
    {
        int o;
        scanf("%d",&o);
        Add(i,o-last);
        last=o;
    }
    for(int i=0;i<m;i++)
    {
        scanf("%d",&q);
        if(q==1)
        {
            scanf("%d%d%d",&w,&e,&r);
            Add(w,r);
            Add(e+1,-r);
        }
        if(q==2)
        {
            scanf("%d",&w);
            printf("%d\n",Sum(w));
        }
    }
    return 0;
}
 

猜你喜欢

转载自blog.csdn.net/to_more_excellent/article/details/81592358
今日推荐