Cod sursa(job #2791366)

Utilizator gripzStroescu Matei Alexandru gripz Data 30 octombrie 2021 13:38:00
Problema Schi Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.72 kb
#include <iostream>
#include <climits>

#define MAXN 30002

using namespace std;

struct schi {
    int pos;
    int val;
    int lazy;
};

schi T[4*MAXN];
int v[MAXN], N;
int pos, val;

void build(int node, int st, int dr) {
    if(st == dr) {
        T[node].pos = st;
        T[node].val = v[st];
        T[node].lazy = 0;
    } else {
        int mid = (st + dr) / 2;
        build(2 * node, st , mid);
        build(2 * node + 1, mid + 1, dr);
        T[node] = T[2 * node + 1];
        if (T[node].val > T[2 * node].val) {
            T[node] = T[2 * node];
        }
    }
}

void propagate(int node, int st, int dr) {
    if(st != dr) {
        T[2 * node].lazy += T[node].lazy;
        T[2 * node + 1].lazy += T[node].lazy;
    }
    T[node].val -= T[node].lazy;
    T[node].lazy = 0;
}

void update(int node, int st, int dr) {
    propagate(node, st, dr);
    if(st == dr) {
        T[node].val = INT_MAX;
    } else {
        int mid = (st + dr) / 2;
        if(pos <= mid) {
            update(2 * node, st, mid);
            T[2 * node + 1].lazy++;
        } else {
            update(2 * node + 1, mid + 1, dr);
        }
        propagate(2 * node, st, mid);
        propagate(2 * node + 1, mid + 1, dr);
        T[node] = T[2 * node + 1];
        if (T[node].val > T[2 * node].val) {
            T[node] = T[2 * node];
        }
    }
}

int main()
{
    freopen("schi.in", "r", stdin);
    freopen("schi.out", "w", stdout);

    cin >> N;

    for(int i = 1; i <= N; i++) {
        cin >> v[i];
    }

    build(1, 1, N);

    for(int i = 1; i <= N; i++) {
        cout << T[1].pos << endl;
        pos = T[1].pos;
        update(1, 1, N);
    }

    return 0;
}