Cod sursa(job #856762)

Utilizator Aida_SilviaStrimbeanu Aida Silvia Aida_Silvia Data 16 ianuarie 2013 21:53:20
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <cstdio>
#include<math.h>
#include<assert.h>
using namespace std;

int m,n,start,finish,pos,el,sum,nr_buckets,bucket_size;
int h[15073],tot[15073];

inline int bucket (int poz) {
    return (poz-1)/bucket_size;
}

void update()
{
    h[pos]+=el;
    tot[bucket (pos)]+=el;
}

void query()
{
    for (int i=start; bucket (start)==bucket (i); ++i)
        sum+=h[i];
    for (int i=bucket (start)+1; i<bucket (finish); ++i)
        sum+=tot[i];
    for (int i=finish; bucket (finish)==bucket (i); --i)
        sum+=h[i];
}


int main()
{
    int x,a,b,op;
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);

    scanf("%d%d",&n,&m);
    bucket_size=(int)sqrt (n);
    nr_buckets=n/bucket_size;
    if (n%bucket_size)
        ++nr_buckets;

    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        pos=i;el=x;
        update();
    }

     for(int i=0;i<m;i++)
     {
         scanf("%d%d%d",&op,&a,&b);
         if (op==1)
         {
             start=a;finish=b;
             sum=0;
             query();
             printf("%d\n",sum);
         }
         else
         {
             pos=a; el=-b;
             update();
         }
     }

     fclose(stdout);
    return 0;
}