Cod sursa(job #2962412)

Utilizator AndreiMargaritMargarit Andrei AndreiMargarit Data 8 ianuarie 2023 15:27:11
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.4 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("aib.in");
ofstream g("aib.out");

int a[100001],v[100001],n,m;

int query(int p)
{
    int suma=0,i;
    for(i=p;i;i=i-(i&-i))
    {
        suma=suma+a[i];
    }
    return suma;
}

void update(int p,int x)
{
    int i;
    for(i=p;i<=n;i=i+(i&-i))
    {
        a[i]=a[i]+x;
    }
}

int main()
{int i,j,x,op,y,s,st,dr,mij,ok,poz;
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        update(i,x);
    }
    for(i=1;i<=m;i++)
    {
        f>>op;
        if(op==0)
        {
            f>>x>>y;
            update(x,y);
        }
        if(op==1)
        {
            f>>x>>y;
            s=query(y)-query(x-1);
            g<<s<<'\n';
        }
        if(op==2)
        {
            f>>x;
            st=1; dr=n; poz=-1;
            while(st<=dr)
            {
                mij=(st+dr)/2;
                s=query(mij);
                if(s==x)
                {
                    poz=mij;
                    dr=mij-1;
                }
                else
                {
                    if(s>x)
                    {
                        dr=mij-1;
                    }
                    else
                    {
                        st=mij+1;
                    }
                }
            }
            g<<poz<<'\n';
        }
    }
    return 0;
}