Cod sursa(job #2690986)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 26 decembrie 2020 17:15:31
Problema Arbori indexati binar Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include<cstdio>
const int N=100001;
int n,m,x,y,z,i,s,c[N],a[N],u,t;
int main()
{
    freopen("aib.in","r",stdin),freopen("aib.out","w",stdout),scanf("%d%d",&n,&m);
    for(s=1;s<=n;s<<=1);
    for(i=1;i<=n;i++)
        scanf("%d",a+i),a[i]+=a[i-1],c[i]=a[i]-a[i-(i&(-i))];
    while(m--)
	{
        scanf("%d%d",&x,&y);
        if(x==2)
        {
            for(t=s,i=0;t;t>>=1)
                if(i+t<=n)
                    if(c[i+t]<=y)
                        i+=t,y-=c[i];
            printf("%d\n",!y?i:-1);
        }
        else
		{
            scanf("%d",&z);
            if(!x)
                for(;y<=n;c[y]+=z,y+=(y&(-y)));
            else
            {
                for(u=t=0;z;u+=c[z],z-=(z&(-z)));
                for(y--;y;t+=c[y],y-=(y&(-y)));
                printf("%d\n",u-t);
            }
        }
    }
}