Cod sursa(job #3233355)

Utilizator cipri05Ciprian Cristescu cipri05 Data 3 iunie 2024 01:47:21
Problema Plantatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <algorithm>
#include <fstream>

using namespace std;

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

int N, M, rmq[17][502][502], putere[502];

int main() {
    fin >> N >> M;

    // Citirea valorilor inițiale în matricea rmq[0]
    for (int i = 1; i <= N; ++i) {
        for (int j = 1; j <= N; ++j) {
            fin >> rmq[0][i][j];
        }
    }

    // Calcularea valorilor pentru restul nivelelor din rmq
    for (int p = 1, dimensiune = 2; dimensiune <= N; p++, dimensiune *= 2) {
        for (int i = 1; i <= N - dimensiune + 1; i++) {
            for (int j = 1; j <= N - dimensiune + 1; j++) {
                int x = i + (dimensiune >> 1), y = j + (dimensiune >> 1);
                rmq[p][i][j] = max(
                    max(rmq[p - 1][i][j], rmq[p - 1][i][y]),
                    max(rmq[p - 1][x][j], rmq[p - 1][x][y])
                );
            }
        }
    }

    // Calcularea valorilor pentru vectorul putere
    putere[0] = 0;
    for (int i = 2; i <= N; ++i) {
        putere[i] = 1 + putere[i / 2];
    }

    // Procesarea interogărilor
    for (int t = 0; t < M; t++) {
        int rand, col, dimensiune;
        fin >> rand >> col >> dimensiune;
        int nivel = putere[dimensiune];
        int lungime = (1 << nivel);

        fout << max(
            max(rmq[nivel][rand][col], rmq[nivel][rand][col + dimensiune - lungime]),
            max(rmq[nivel][rand + dimensiune - lungime][col], rmq[nivel][rand + dimensiune - lungime][col + dimensiune - lungime])
        ) << "\n";
    }

    return 0;
}