Cod sursa(job #3324313)

Utilizator alex.iovita.23@gmail.comIovita Alexandru [email protected] Data 22 noiembrie 2025 02:05:02
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.06 kb
#include<bits/stdc++.h>
#define DIM 100100

using namespace std;

ifstream fin("datorii.in");
ofstream fout("datorii.out");

int n , m , s;

struct Iris{
    int val , flag;
}A[4 * DIM];

inline void build(int nod , int st ,int dr){
    if(st == dr){
        fin >> A[nod].val;
        A[nod].flag = 0;
    }
    else{
        int mid = (st + dr) / 2;
        build(2 * nod , st , mid);
        build(2 * nod + 1 , mid + 1 , dr);
        A[nod].val = A[2 * nod].val + A[2 * nod + 1].val;
    }
}

inline void query(int nod , int st , int dr , int a , int b){
    if(a <= st && dr <= b){
        s += A[nod].val;
    }
    else{
        if(A[nod].flag == 1){
            A[2 * nod].val = A[nod].val;
            A[2 * nod + 1].val = A[nod].val;
            A[2 * nod].flag = 1;
            A[2 * nod + 1].flag = 1;
            A[nod].flag = 0;
        }
        int mid = (st + dr) / 2;
        if(a <= mid)
            query(2 * nod , st , mid , a , b);
        if(b > mid)
            query(2 * nod + 1, mid + 1 , dr , a , b);
        A[nod].val = A[2 * nod].val + A[2 * nod + 1].val;
    }
}

inline void update(int nod , int st , int dr , int poz , int x){
    if(st == dr){
        A[nod].val -= x;
    }
    else{
        if(A[nod].flag == 1){
            A[2 * nod].val = A[nod].val;
            A[2 * nod + 1].val = A[nod].val;
            A[2 * nod].flag = 1;
            A[2 * nod + 1].flag = 1;
            A[nod].flag = 0;
        }
        int mid = (st + dr) / 2;
        if(poz <= mid)
            update(2 * nod , st , mid , poz , x);
        if(poz > mid)
            update(2 * nod + 1 , mid + 1 , dr , poz , x);
        A[nod].val = A[2 * nod].val + A[2 * nod + 1].val;
    }
}

int main(){
    fin >> n >> m;
    build(1 , 1 , n);
    for(int i = 1 ; i <= m ; i++){
        int t , a , b;
        fin >> t >> a >> b;
        if(t == 1){
            s = 0;
            query(1 , 1 , n , a , b);
            fout << s << '\n';
        }
        else{
            update(1 , 1 , n , a , b);
        }
    }
}