Cod sursa(job #1974150)

Utilizator pepsiM4A1Ozturk Arif pepsiM4A1 Data 26 aprilie 2017 22:39:35
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.11 kb
//try with sqrt
#include <cstdio>
#include <cmath>
using namespace std;
const int DIM=50000,nmax=100003,bmax=317;
int poz=DIM-1;
char buff[DIM];
void citeste(int &nr)
{
    nr=0;
    while(!('0'<=buff[poz]&&buff[poz]<='9'))
    {
        if(++poz==DIM) fread(buff,1,DIM,stdin),poz=0;
    }
    while('0'<=buff[poz]&&buff[poz]<='9')
    {
        nr=nr*10 + buff[poz]-'0';
        if(++poz==DIM) fread(buff,1,DIM,stdin),poz=0;
    }
}
int v[nmax],s[bmax];
unsigned char piece[nmax];
int n,m,x;
inline void citire()
{
    citeste(n),citeste(m);
    x=sqrt(n);
    for(int i=1;i<=n;i++) citeste(v[i]);
    int pieceT=0;
    for(int i=1;i<=n;i+=x)
    {
        ++pieceT;
        for(int j=i;j<i+x&&j<=n;j++)
        {
            piece[j]=pieceT;
            s[pieceT]+=v[j];
        }
    }
}
int query1(int p1,int p2)
{
    int sum=0;
    for(int i=p1;i<=p2;i++)
    {
        if(piece[i]!=piece[i-1]&&i+x-1<=p2)
        {
            sum+=s[piece[i]];
            i+=x;
            --i;
        }
        else sum+=v[i];
    }
    return sum;
}
int query2(int val)
{
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        if(piece[i]!=piece[i-1]&&i+x-1<=n&&sum+s[piece[i]]<=val)
        {
            sum+=s[piece[i]];
            i+=x;
            --i;
            if(sum==val) return i;
        }
        else
        {
            sum+=v[i];
            if(sum==val) return i;
            else if(sum>val) return -1;
        }
    }
    return -1;
}
inline void solve()
{
    for(;m>0;--m)
    {
        int tip;
        citeste(tip);
        if(tip==0)
        {
            int a,b;
            citeste(a),citeste(b);
            v[a]+=b;
            s[piece[a]]+=b;
        }
        else if(tip==1)
        {
            int a,b;
            citeste(a),citeste(b);
            printf("%d\n",query1(a,b));
        }
        else
        {
            int a;
            citeste(a);
            printf("%d\n",query2(a));
        }
    }
}
int main()
{
    freopen ("aib.in","r",stdin);
    freopen ("aib.out","w",stdout);
    citire();
    solve();
}