Cod sursa(job #2438041)

Utilizator AndreiD31Dragan Andrei AndreiD31 Data 11 iulie 2019 08:59:10
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <bits/stdc++.h>
#define bits(i) (i&(-i))
using namespace std;

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

int n,AIB[100100];

void adaug(int poz,int val)
{
    for(int i=poz;i<=n;i+=bits(i))
        AIB[i]+=val;
}

int suma(int poz)
{
    int S=0;
    for(int i=poz;i>=1;i-=bits(i))
        S+=AIB[i];
    return S;
}


int m,cerinta,i,x,a,b,cpy_poz,p,u,mij;
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        adaug(i,x);
    }

    for(int k=1;k<=m;k++)
    {
        f>>cerinta;
        if(cerinta==1)
        {
            f>>a>>b;
            g<<suma(b)-suma(a-1)<<'\n';
        }
        else if(cerinta==2)
        {
            f>>a;
            p=1;u=n;
            while(p<=u)
            {
                mij=(p+u)/2;
                if(suma(mij)>a){u=mij-1;}
                else {cpy_poz=mij; p=mij+1;}
            }
            //g<<cpy_poz<<" "<<suma(cpy_poz)<<"  ";
            if(suma(cpy_poz)==a)g<<cpy_poz<<'\n';
            else g<<-1<<'\n';
        }
        else
        {
            f>>a>>b;
            adaug(a,b);
        }
    }
    return 0;
}