Pagini recente » Cod sursa (job #908261) | Cod sursa (job #717792) | Cod sursa (job #1879466) | Cod sursa (job #281858) | Cod sursa (job #438096)
Cod sursa(job #438096)
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#include <stdint.h>
#define INDEX(h) (((H)/(U)) - ((H - (h))/U)) // nivelul in care intra o anumita gutaie in functie de inaltime
#define LEVELS ((H/U) + 1) // numarul de nivele in functie de inaltimea maxima si diferenta dintre
// nivele
// Macrouri pentru heap-uri
#define PARENT(id) ((id + 1)/2 - 1)
#define LEFT(id) ((id)*2 + 1)
#define RIGHT(id) ((id)*2 + 2)
using namespace std;
uint32_t N, H, U; // Numarul de Gutui, Inaltimea maxima, Pasul de inaltare
ifstream in ("gutui.in");
ofstream out ("gutui.out");
typedef struct _iter_info
{
uint32_t max;
uint32_t index;
} IterInfo, *PIterInfo;
class Level {
private:
vector<uint32_t> data;
uint32_t crt_max;
public:
void add_gut(uint32_t g);
void partial_sort_(uint32_t n);
uint32_t max();
uint32_t get_max();
};
class LevelHeap {
private:
uint32_t size;
Level* data;
void bring_up(uint32_t index);
void bring_down(uint32_t index);
public:
LevelHeap(Level* level_data);
void add_level();
uint32_t get_max();
};
bool compare_g(uint32_t g1, uint32_t g2) { return (g1 < g2); }
bool compare_lvl(Level l1, Level l2) { return (l1.max() < l2.max()); }
int main()
{
in >> N >> H >> U;
uint32_t sum = 0;
const uint32_t max_sorted = ( N > LEVELS ) ? LEVELS : N;
uint32_t temp_g, temp_h;
Level* levels = new Level[LEVELS];
for(uint32_t i = 0; i < N; i ++)
{
in >> temp_h >> temp_g;
levels[INDEX(temp_h)].add_gut(temp_g);
}
for(uint32_t i; i < LEVELS; i ++)
levels[i].partial_sort_(max_sorted - i);
/* PIterInfo iter_info = new IterInfo[LEVELS];
uint32_t temp_max;
iter_info[0].max = levels[0].max();
iter_info[0].index = 0;
for(uint32_t i = 0; i < LEVELS; i ++)
{
temp_max = levels[i].max();
if(temp_max > iter_info[i - 1].max)
{
iter_info[i].max = temp_max;
iter_info[i].index = i;
}
else
{
iter_info[i].max = iter_info[i - 1].max;
iter_info[i].index = iter_info[i - 1].index;
}
}*/
LevelHeap level_heap(levels);
for(uint32_t i = 0; i < LEVELS; i ++)
{
level_heap.add_level();
sum += level_heap.get_max();
}
out << sum;
return 0;
}
void Level::add_gut(uint32_t g)
{
data.push_back(g);
}
void Level::partial_sort_(uint32_t n)
{
partial_sort(data.begin(), data.begin() + n, data.end(), compare_g);
crt_max = 0;
}
uint32_t Level::max()
{
if(data.size() == 0)
return 0;
return data[0];
}
uint32_t Level::get_max()
{
if(data.size() == 0)
return 0;
return data[crt_max];
crt_max ++;
}
LevelHeap::LevelHeap(Level* level_data)
{
data = level_data;
size = 0;
}
void LevelHeap::add_level()
{
size ++;
bring_up(size - 1);
}
uint32_t LevelHeap::get_max()
{
uint32_t max = data[0].get_max();
bring_down(0);
return max;
}
void LevelHeap::bring_up(uint32_t index)
{
if(index <= 0)
return;
if(data[index].max() > data[PARENT(index)].max() )
{
Level temp = data[index];
data[index] = data[PARENT(index)];
data[PARENT(index)] = data[index];
bring_up(PARENT(index));
}
}
void LevelHeap::bring_down(uint32_t index)
{
uint32_t max;
uint32_t left = LEFT(index), right = RIGHT(index);
if(left <= size && data[left].max() > data[index].max())
max = left;
else
max = index;
if(right <= size && data[right].max() > data[max].max())
max = right;
if(max != index)
{
Level temp = data[max];
data[max] = data[index];
data[index] = temp;
bring_down(max);
}
}