Cod sursa(job #905906)

Utilizator valentina506Moraru Valentina valentina506 Data 6 martie 2013 11:59:39
Problema Arbori indexati binar Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.34 kb
#include<fstream>
using namespace std;
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 aduna(int poz)
{
    int s=0;
    while(poz!=0)
    {
        s+=c[poz];
        poz-=(poz&-poz);
    }
    return s;
}
int gaseste(int s)
{
    int st=1,dr=n,mij,sum=0;
    while(st<dr)
    {
        mij=(st+dr)/2;
        sum=aduna(mij);
        if(sum==s)
            return mij;
            if(sum<s)
                st=mij+1;
            else
                if(sum>s)
                dr=mij-1;
    }
    if(aduna(st)==s)
        return st;
    return -1;
}
int main()
{
    ifstream f("aib.in");
    ofstream g("aib.out");
    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=aduna(a1-1);
                s2=aduna(a2);
                g<<s2-s1<<"\n";
            }
            else
            {
                f>>s;
                g<<gaseste(s)<<"\n";
            }
    }
    return 0;
}