Cod sursa(job #1639990)

Utilizator SlevySlevoaca Stefan-Gabriel Slevy Data 8 martie 2016 15:11:57
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#define lsb(x) ((x^(x-1))&x)

using namespace std;

const int NMAX = 15001;
int n,m;
int AIB[NMAX];

void Add(int day,int money)
{
    for(int i = day;i<=n;i+=lsb(i))
        AIB[i]+=money;
}

void read()
{
    scanf("%d %d",&n,&m);
    int x;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        Add(i,x);
    }
}

void Pay(int day,int money)
{
    for(int i=day;i<=n;i+=lsb(i))
        AIB[i]-=money;
}

int Get_Debt_Ini(int day)
{
    int money = 0;
    for(int i = day;i>0;i-=lsb(i))
        money += AIB[i];
    return money;
}

int Get_Debt(int day1,int day2)
{
    if(day1>day2)
        swap(day1,day2);
    return Get_Debt_Ini(day2)-Get_Debt_Ini(day1-1);
}

int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    read();
    int q,a,b;
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d %d",&q,&a,&b);
        if(q==0)
            Pay(a,b);
        else
            printf("%d\n",Get_Debt(a,b));
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}