Cod sursa(job #3256439)

Utilizator Silviu643Dumitrescu Silviu Florian Silviu643 Data 14 noiembrie 2024 16:38:04
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n,aib[100005],m;
void adun(int x,int val)
{
    for(int i=x;i<=n;i+=(i&(-i)))
    {
        aib[i]+=val;
    }
}
int suma(int x)
{
    int sum=0,i;
    for(int i=x;i>=1;i-=(i&(-i)))
        sum+=aib[i];
        return sum;
}
int main()
{
    fin>>n>>m;
    int nr;
    for(int i=1;i<=n;i++)
        {fin>>nr; adun(i,nr);}
    for(int i=1;i<=m;i++)
    {
        short cer;
        fin>>cer;
        if(cer==0)
        {
            int poz1,nr1;
            fin>>poz1>>nr1;
            adun(poz1,nr1);
        }
        if(cer==1)
        {
            int x1,x2;
            fin>>x1>>x2;
            fout<<suma(x2)-suma(x1-1)<<'\n';
        }
        if(cer==2)
        {
            int st=1,dr=n,v;
            fin>>v;
           int p=1,u=n,mij,ok=0;
    while(p<=u && !ok)
    {mij=(p+u)/2;
    if(suma(mij)<v)p=mij+1;
    else if(suma(mij)>v)u=mij-1;
    else ok=1;
    }
    if(ok)fout<<mij<<'\n';
    else fout<<-1<<'\n';

        }
    }

    return 0;
}