Cod sursa(job #2818604)

Utilizator rares89_Dumitriu Rares rares89_ Data 16 decembrie 2021 16:16:39
Problema SequenceQuery Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.73 kb
#include <fstream>
#define lli long long int

using namespace std;

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

struct node {
    lli suma;
    lli prefsmax, suffsmax;
    lli smax;
} tree[400005];

int n, v[100005];

node update(node st, node dr) {
    node ans;
    ans.suma = st.suma + dr.suma;
    ans.prefsmax = max(st.prefsmax, st.suma + dr.prefsmax);
    ans.suffsmax = max(dr.suffsmax, dr.suma + st.suffsmax);
    ans.smax = max(st.suffsmax + dr.prefsmax, max(st.smax, dr.smax) );
    return ans;
}

void build(int nod, int st, int dr) {
    if(st == dr) {
        tree[nod].suma = v[st];
        tree[nod].prefsmax = v[st];
        tree[nod].suffsmax = v[st];
        tree[nod].smax = v[st];
    } else {
        int mid = (st + dr) / 2;
        
        build(2 * nod, st, mid);
        build(2 * nod + 1, mid + 1, dr);
        
        tree[nod] = update(tree[2 * nod], tree[2 * nod + 1]);
        
    }
}

node query(int nod, int st, int dr, int x, int y) {
    if(st >= x && dr <= y) {
        return tree[nod];
    } else {
        int mid = (st + dr) / 2;
        if(y <= mid) {
            return query(2 * nod, st, mid, x, y);
        }
        if(mid + 1 <= x) {
            return query(2 * nod + 1, mid + 1, dr, x, y);
        }
        node left = query(2 * nod, st, mid, x, y);
        node right = query(2 * nod + 1, mid + 1, dr, x, y);
        return update(left, right);
    }
}

int main() {
    int q;
    fin >> n >> q;
    for(int i = 1; i <= n; i++) {
        fin >> v[i];
    }
    build(1, 1, n);
    for(int i = 1; i <= q; i++) {
        int x, y;
        fin >> x >> y;
        fout << query(1, 1, n, x, y).smax << "\n";
    }
    return 0;
}