Cod sursa(job #1301780)

Utilizator horiainfoTurcuman Horia horiainfo Data 26 decembrie 2014 13:18:48
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.31 kb
#include <fstream>
#include <cstdio>
using namespace std;
ofstream fout("datorii.out");
int n,m;
struct A{int dat,poz;} v[15001];
struct arbore{int inc,sf,dat;} nod[40000];
int create(int inc,int sf,int poz)
{
    nod[poz].inc=inc,nod[poz].sf=sf;
    if(inc==sf)
    {
        nod[poz].dat=v[inc].dat;
        v[inc].poz=poz;
    }
    else
        nod[poz].dat+=create(inc,(inc+sf)/2,poz*2)+create((inc+sf)/2+1,sf,poz*2+1);
    return nod[poz].dat;
}
void achit(int poz,int sum)
{
    while(poz)
        nod[poz].dat-=sum,poz/=2;
}
int query(int poz,int inc,int sf)
{
    int mij=(nod[poz].inc+nod[poz].sf)/2;
    if(inc==nod[poz].inc && sf==nod[poz].sf)
        return nod[poz].dat;
    else
        if(sf<=mij)
            return query(poz*2,inc,sf);
        else
            if(inc>mij)
                return query(poz*2+1,inc,sf);
            else
                return query(poz*2,inc,mij)+query(poz*2+1,mij+1,sf);
}
int main()
{
    int x,y,aux;
    freopen("datorii.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&v[i].dat);
    create(1,n,1);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&aux,&x,&y);
        if(aux==0)
            achit(v[x].poz,y);
        else
            fout<<query(1,x,y)<<'\n';
    }
    return 0;
}