Pagini recente » Cod sursa (job #2633832) | Cod sursa (job #3031909) | Cod sursa (job #139322) | Cod sursa (job #1615171) | Cod sursa (job #2811347)
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>
#define MAX_SIZE 100001
struct Sheep {
private:
size_t wool = 0;
public:
size_t distance = 0, max_step = 0;
size_t getWool() const {
return wool;
}
Sheep() = default;
Sheep(std::istream &istream, size_t wolf_max_distance, size_t 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<Sheep, std::vector<Sheep>> woolQueue;
size_t n, wolf_max_distance, increment, max_step = 0;
Sheep sheep[MAX_SIZE];
input >> n >> wolf_max_distance >> increment;
for (size_t 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);
size_t index = 0;
size_t total_wool = 0;
for (size_t i = max_step; i >= 1; --i) {
while (index < n && sheep[index].max_step == i) woolQueue.push(sheep[index++]);
if (woolQueue.size() > 0) {
total_wool += woolQueue.top().getWool();
woolQueue.pop();
}
}
output << total_wool;
return 0;
}