Cod sursa(job #2602852)

Utilizator lev.tempfliTempfli Levente lev.tempfli Data 17 aprilie 2020 23:59:27
Problema Plantatie Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.3 kb
//https://infoarena.ro/preoni-2007/runda-2/solutii
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>

template<class T>
class Range_minimum_query {
public:
    explicit Range_minimum_query(std::vector<std::vector<T>> &arr);

    T query(int i, int j, int k);

private:
    std::vector<std::vector<std::vector<T>>> lookup;
    std::vector<int> lg2;
};

template<class T>
Range_minimum_query<T>::Range_minimum_query(std::vector<std::vector<T>> &arr) {
    lookup.resize(log2(arr.size()) + 1, std::vector<std::vector<T>>(arr.size(), std::vector<T>(arr.size())));
    for (int i = 0; i < arr.size(); i++) {
        for (int j = 0; j < arr.size(); j++) {
            lookup[0][i][j] = arr[i][j];
        }

    }
    for (int k = 1; (1 << k) <= arr.size(); k++) {
        for (int i = 0; i <= arr.size() - (1 << k); i++) {
            for (int j = 0; j <= arr.size() - (1 << k); j++) {
                lookup[k][i][j] = std::max(std::max(lookup[k - 1][i][j], lookup[k - 1][i + (1 << k - 1)][j]),
                                           std::max(lookup[k - 1][i][j + (1 << k - 1)],
                                                    lookup[k - 1][i + (1 << k - 1)][j + (1 << k - 1)]));
            }

        }

    }
    lg2.resize(arr.size());
    for (int i = 2; i < arr.size(); i++) {

        lg2[i] = lg2[i / 2] + 1;

    }
}

template<class T>
T Range_minimum_query<T>::query(int i, int j, int k) {
    //int lg = log2(j - i + 1);
    int t1 = lookup[lg2[k]][i][j];
    int t2 = lookup[lg2[k]][i + k - (1 << lg2[k])][j + k - (1 << lg2[k])];
    int t3 = lookup[lg2[k]][i][j + k - (1 << lg2[k])];
    int t4 = lookup[lg2[k]][i + k - (1 << lg2[k])][j];

    return std::max(std::max(t1, t2), std::max(t3, t4));
}

int main() {
    std::ifstream fin("plantatie.in");


    std::vector<std::vector<int>> a;
    int n, m;
    fin >> n >> m;
    a.resize(n, std::vector<int>(n, 0));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            int g;
            fin >> g;
            std::cout << g << " ";
            a[i][j] = g;
        }
        std::cout << "\n";
    }


    std::ofstream fout("plantatie.out");
    Range_minimum_query<int> rmq(a);
    int f, g, h;
    for (int i = 0; i < m; i++) {
        fin >> f >> g >> h;
        fout << rmq.query(f - 1, g - 1, h) << "\n";
    }

    return 0;
}