Cod sursa(job #1334740)

Utilizator bogdanboboc97Bogdan Boboc bogdanboboc97 Data 4 februarie 2015 16:49:47
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in("datorii.in");
ofstream out("datorii.out");
int tree[400004],pos,y,a,b,sum,n,m,t;
void update(int x,int left,int right)
{
    if(left==right)tree[x]+=y;
    else{
        int mid=(left+right)/2;
        if(pos<=mid)
        {
            update(2*x,left,mid);
            tree[x]+=y;
        }
        else
        {
            update(2*x+1,mid+1,right);
            tree[x]+=y;
        }
    }
}

void query(int x,int left,int right)
{
    if(a<=left && b>=right)sum+=tree[x];
    else{
        int mid=(left+right)/2;
        if(a<=mid)query(2*x,left,mid);
        if(b>mid)query(2*x+1,mid+1,right);
    }
}
int main()
{
    in>>n>>m;
    for(int i=1;i<=n;i++)
    {
        in>>y;
        pos=i;
        update(1,1,n);
    }
    for(;m;m--)
    {
        in>>t>>a>>b;
        if(t==1)
        {
            sum=0;
            query(1,1,n);
            out<<sum<<'\n';
        }
        else{
            pos=a;y=-b;
            update(1,1,n);
        }
    }
    return 0;
}