Cod sursa(job #3233510)

Utilizator Mihai_OctMihai Octavian Mihai_Oct Data 3 iunie 2024 18:19:55
Problema Plantatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.18 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("plantatie.in");
ofstream fout("plantatie.out");
int n, q, p, i, j, rmq[10][502][502];
int lg[502];

static inline int Max(int i, int j, int k) {
    int l = lg[k];
    int p = (1 << l);
    return max(max(rmq[l][i][j], rmq[l][i + k - p][j + k - p]),
               max(rmq[l][i + k - p][j], rmq[l][i][j + k - p]));
}

int main() {
    ios_base::sync_with_stdio(false);
    fin.tie(nullptr);
    fout.tie(nullptr);

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

    for(i = 2; i <= n; i++) lg[i] = 1 + lg[i >> 1];

    for(int p = 1; p <= lg[n]; p++) {
        int l = (1 << (p - 1));
        for(int i = 1; i + (1 << p) - 1 <= n; i++) {
            for(int j = 1; j + (1 << p) - 1 <= n; j++)
                rmq[p][i][j] = max(max(rmq[p - 1][i    ][j], rmq[p - 1][i + l][j + l]),
                                   max(rmq[p - 1][i + l][j], rmq[p - 1][i    ][j + l]));
        }
    }

    while(q--) {
        int x, y, k;
        fin >> x >> y >> k;
        fout << Max(x, y, k) << "\n";
    }

    return 0;
}