Cod sursa(job #2461807)

Utilizator HannaLieb Hanna Hanna Data 26 septembrie 2019 10:21:02
Problema Arbori indexati binar Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.88 kb
#include <fstream>
#include <vector>

using namespace std;

ifstream cin("aib.in");
ofstream cout("aib.out");

int n,m,i,t;
long long a,b;
vector<long long>x,ossz;
vector<long long>:: iterator it;

int main()
{
    cin>>n>>m;
    x.resize(n+1,0);
    ossz.resize(n+1,0);
    for(i=1;i<=n;++i)
    {
        cin>>x[i];
        ossz[i]=ossz[i-1]+x[i];
    }
    while(m)
    {
        cin>>t;
        if(t==0)
        {
            cin>>a>>b;
            x[a]+=b;
            for(i=a;i<=n;++i) ossz[i]+=b;
        }
        else if(t==1)
        {
            cin>>a>>b;
            cout<<ossz[b]-ossz[a-1]<<"\n";
        }
        else
        {
            cin>>a;
            it=lower_bound(ossz.begin(), ossz.end(), a);
            if(*it==a) cout<<it-ossz.begin()<<"\n";
            else cout<<"-1\n";
        }
        --m;
    }

    return 0;
}