Cod sursa(job #2592978)

Utilizator copanelTudor Roman copanel Data 2 aprilie 2020 17:52:04
Problema Plantatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.23 kb
#include <fstream>
#include <algorithm>

int rmq[10][501][501];
int log2[501];

int main() {
    std::ifstream fin("plantatie.in");
    std::ofstream fout("plantatie.out");
    int n, m;

    fin >> n >> m;
    log2[1] = 0;
    for (int i = 2; i <= n; i++) {
        log2[i] = 1 + log2[i / 2];
    }

    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            fin >> rmq[0][i][j];
        }
    }

    int log = log2[n];
    for (int i = 1; i <= log; i++) {
        int half = (1 << (i - 1));
        for (int j = (1 << i); j <= n; j++) {
            for (int k = (1 << i); k <= n; k++) {
                rmq[i][j][k] = std::max({
                    rmq[i - 1][j][k],        rmq[i - 1][j][k - half],
                    rmq[i - 1][j - half][k], rmq[i - 1][j - half][k - half]
                });
            }
        }
    }

    for (; m > 0; m--) {
        int a, b, latura;
        fin >> a >> b >> latura;
        a += latura - 1;
        b += latura - 1;
        int l = log2[latura];
        int power = 1 << l;
        fout << std::max({
            rmq[l][a][b], rmq[l][a][b - latura + power],
            rmq[l][a - latura + power][b], rmq[l][a - latura + power][b - latura + power]
        }) << '\n';
    }
    return 0;
}