Cod sursa(job #971817)

Utilizator monica11Szekely Monica monica11 Data 10 iulie 2013 12:05:17
Problema Arbori indexati binar Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.34 kb
#include<fstream>
using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");
long i,j,n,m,a[100010],c[100010],tip,a1,a2,s1,s2,s;
void schimba(int poz,int x)
{
    while(poz<=n)
    {
        c[poz]+=x;
        poz+=(poz&-poz);
    }
}
int sumint(int poz)
{
    int s=0;
    while(poz!=0)
    {
        s+=c[poz];
        poz-=(poz&-poz);
    }
    return s;
}
int detpoz(int s)
{
    int st=1,dr=n,mij,sum=0;
    while(st<dr)
    {
        mij=(st+dr)/2;
        sum=sumint(mij);
        if(sum==s)
            return mij;
            if(sum<s)
                st=mij+1;
            else
                if(sum>s)
                dr=mij-1;
    }
    if(sumint(st)==s)
        return st;
    return -1;
}
int main()
{
    f>>n>>m;
    a[0]=0;
    for(i=1;i<=n;++i)
    {
        f>>a[i];
        a[i]+=a[i-1];
        c[i]=a[i]-a[i-(i&-i)];
    }
    for(i=1;i<=m;++i)
    {
        f>>tip;
        if(tip==0)
        {
            f>>a1>>a2;
            schimba(a1,a2);
        }
        else
            if(tip==1)
            {
                f>>a1>>a2;
                s1=sumint(a1-1);
                s2=sumint(a2);
                g<<s2-s1<<"\n";
            }
            else
            {
                f>>s;
                g<<detpoz(s)<<"\n";
            }
    }
    return 0;
}