Cod sursa(job #2786328)

Utilizator LucaMihaiLM10Luca Ilie LucaMihaiLM10 Data 20 octombrie 2021 18:32:36
Problema Datorii Scor 100
Compilator c-64 Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <stdio.h>

#define MAX_N 100000

int aib[MAX_N + 1];

void update( int i, int a ) {
    while ( i <= MAX_N ) {
        aib[i] += a;
        i += i & -i;
    }
}

int query( int i ) {
    int s;

    s = 0;
    while ( i > 0 ) {
        s += aib[i];
        i -= i & -i;
    }

    return s;
}

int main() {
    FILE *fin, *fout;
    int n, q, tip, a, s, f, p, i;

    fin = fopen( "datorii.in", "r" );

    fscanf( fin, "%d%d", &n, &q );
    for ( i = 1; i <= n; i++ ) {
        fscanf( fin, "%d", &a );
        update( i, a );
    }

    fout = fopen( "datorii.out", "w" );

    for ( i = 0; i < q; i++ ) {
        fscanf( fin, "%d", &tip );
        if ( tip == 0 ) {
            fscanf( fin, "%d%d", &p, &a );
            update( p, -a );
        }
        else {
            fscanf( fin, "%d%d", &s, &f );
            fprintf( fout, "%d\n", query( f ) - query( s - 1 ) );
        }
    }

    fclose( fin );
    fclose( fout );

    return 0;
}