Cod sursa(job #2501416)

Utilizator preda.andreiPreda Andrei preda.andrei Data 29 noiembrie 2019 18:10:26
Problema Range minimum query Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.5 kb
#include <fstream>
#include <vector>

template <typename T>
class RmqVector
{
public:
    RmqVector(int size) {
        mins_.resize(size, {T()});
        logs_.resize(size + 1, 0);

        for (int i = 2; i <= size; i += 1) {
            logs_[i] = logs_[i / 2] + 1;
        }
    }

    void set(int pos, const T &value) {
        mins_[pos][0] = value;
    }

    T get(int left, int right) const {
        int len = right - left + 1;
        int log = logs_[len];

        return std::min(mins_[right][log], mins_[left + (1 << log) - 1][log]);
    }

    void preprocess() {
        for (size_t i = 1; i < mins_.size(); i += 1) {
            int log = 0;
            int mid = i - 1;

            while (mid >= 0 && log < (int)mins_[mid].size()) {
                auto new_min = std::min(mins_[i][log], mins_[mid][log]);
                mins_[i].push_back(new_min);
                log += 1;
                mid = i - (1 << log);
            }
        }
    }

private:
    std::vector<std::vector<T>> mins_;
    std::vector<int> logs_;
};

int main() {
    std::ifstream fin("rmq.in");
    std::ofstream fout("rmq.out");

    int size, queries;
    fin >> size >> queries;

    RmqVector<int> rmq(size);
    for (int i = 0; i < size; i += 1) {
        int num;
        fin >> num;
        rmq.set(i, num);
    }
    rmq.preprocess();

    for (int i = 0; i < queries; i += 1) {
        int left, right;
        fin >> left >> right;

        auto res = rmq.get(left - 1, right - 1);
        fout << res << "\n";
    }

    return 0;
}