Cod sursa(job #2811356)

Utilizator AleXutzZuDavid Alex Robert AleXutzZu Data 1 decembrie 2021 21:35:22
Problema Lupul Urias si Rau Scor 84
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.64 kb
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>

#define MAX_SIZE 100000

struct Sheep {
private:
    unsigned long long wool = 0;
public:
    unsigned long long distance = 0, max_step = 0;

    unsigned long long getWool() const {
        return wool;
    }

    Sheep() = default;

    Sheep(std::istream &istream, unsigned long long wolf_max_distance, unsigned long long increment) {
        istream >> distance >> wool;
        max_step = (wolf_max_distance - distance) / increment + 1;
    }

    static bool cmp_by_step(const Sheep &_lhs, const Sheep &_rhs) {
        return _lhs.max_step > _rhs.max_step;
    }
};

inline bool operator<(const Sheep &_lhs, const Sheep &_rhs) {
    return _lhs.getWool() < _rhs.getWool();
}

int main() {
    std::ifstream input("lupu.in");
    std::ofstream output("lupu.out");

    std::priority_queue<unsigned long long> woolQueue;
    unsigned long long n, wolf_max_distance, increment, max_step = 0;
    Sheep sheep[MAX_SIZE];
    input >> n >> wolf_max_distance >> increment;
    for (unsigned long long i = 0; i < n; ++i) {
        sheep[i] = Sheep(input, wolf_max_distance, increment);
        if (sheep[i].max_step > max_step) max_step = sheep[i].max_step;
    }
    std::sort(sheep, sheep + n, Sheep::cmp_by_step);
    unsigned long long index = 0;
    unsigned long long total_wool = 0;
    for (unsigned long long i = max_step; i >= 1; --i) {
        while (index < n && sheep[index].max_step == i) woolQueue.push(sheep[index++].getWool());

        if (!woolQueue.empty()) {
            total_wool += woolQueue.top();
            woolQueue.pop();
        }
    }
    output << total_wool;
    return 0;
}