Cod sursa(job #3122848)

Utilizator Shaan_StefanShaan Stefan Shaan_Stefan Data 20 aprilie 2023 19:49:56
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.46 kb
#include <fstream>
#define ldigit(x) (x^(x-1))&x
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n, m, i, k, a[100010], c, x, nr, s1, s2, bit[100010], st, dr, mij;
int main()
{
    fin>>n>>m;
    for(i=1; i<=n; i++)
    {
        fin>>a[i];
        for(k=i; k<=n; k+=ldigit(k))
            bit[k]+=a[i];
    }

    for(i=1; i<=m; i++)
    {
        fin>>c>>x;
        if(c==0)
        {
            fin>>nr;
            for(k=x; k<=n; k+=ldigit(k))
            {
                bit[k]+=nr;
            }
        }
        if(c==1)
        {
            fin>>nr;
            s1=0;
            s2=0;
            for(k=x-1; k>0; k-=ldigit(k))
                s1+=bit[k];
            for(k=nr; k>0; k-=ldigit(k))
                s2+=bit[k];
            fout<<s2-s1<<'\n';
        }
        if(c==2)
        {
            st=1, dr=n;
            while(st<=dr)
            {
                s1=0;
                mij=(st+dr)/2;
                for(k=mij; k>0; k-=ldigit(k))
                    s1+=bit[k];
                if(s1==x)
                {
                    fout<<mij<<'\n';
                    break;
                }
                else
                {
                    if(s1<x)
                        st=mij+1;
                    else
                        dr=mij-1;
                }
            }
            if(st>dr)
                fout<<"-1\n";
        }
    }
    return 0;
}