Cod sursa(job #1807418)

Utilizator adystar00Bunea Andrei adystar00 Data 16 noiembrie 2016 16:18:40
Problema Arbori indexati binar Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 2.37 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#include <cstring>
#include <cmath>
using namespace std;
int n,v[400010],aib[400010],s[4000010],answer;
const int INF = 0;
void recalculate(int node)
{
    aib[node]=aib[2*node]+aib[2*node+1];
}
void build (int node, int left, int right)
{
    if(left==right)
        aib[node]=v[left];
    else
    {
        int mij = (left+right)/2;
        build(2*node,left,mij);
        build(2*node+1,mij+1,right);
        recalculate(node);
    }
}
void update (int node, int left, int right,int x, int y)
{
    if(right==left)
        aib[node]+=y;
    else
    {
        int mij = (left+right)/2;
        if(mij>=x)
            update(2*node,left,mij,x,y);
        else
            update(2*node+1,mij+1,right,x,y);
        recalculate(node);
    }
}
int query (int node,int left, int right, int x, int y)
{
    if(x<=left && y>=right)
        return aib[node];
    else
    {
        int answer = 0;
        int mij = (left+right)/2;
        if(x<=mij)
            answer = answer+query(2*node,left,mij,x,y);
        if(y>=mij+1)
            answer = answer+query(2*node+1,mij+1,right,x,y);
        return answer;
    }
}
int main()
{
    ifstream fin ("aib.in");
    ofstream fout ("aib.out");
    int m,i,a,q,x,y,ans;
    fin>>n>>m;
    for(i=1; i<=n; i++)
        fin>>v[i];
    build(1,1,n);
    for(q=1; q<=m; q++)
    {
        fin>>a;
        if(a==0)
        {
            fin>>x>>y;
            v[x]+=y;
            update(1,1,n,x,y);
        }
        if(a==1)
        {
            fin>>x>>y;
            answer = 0;
            fout<<query(1,1,n,x,y)<<"\n";
        }
        if(a==2)
        {
            fin>>x;
            ans=-1;
            s[1]=v[1];
            for(i=2; i<=n; i++)
                s[i]=s[i-1]+v[i];
            int st=1;
            int dr=n;
            int mij = 0;
            while(st<=dr)
            {
                mij = (st+dr)/2;
                if(x<s[mij])
                    dr=mij-1;
                else
                {
                    if(x==s[mij])
                    {
                        ans = mij;
                        st=dr+1;
                    }
                    st = mij+1;
                }
            }
            fout<<ans<<"\n";
        }
    }
    return 0;
}