Cod sursa(job #442059)

Utilizator emilia.ciobanuCiobanu Emilia Maria Silvia emilia.ciobanu Data 13 aprilie 2010 20:35:45
Problema Gutui Scor 100
Compilator cpp Status done
Runda teme_upb Marime 1.49 kb
#include<fstream>
#include<cstdlib>
#include<climits>
#include<vector>
#include<algorithm>

#define IN "gutui.in"
#define OUT "gutui.out"

using namespace std;

typedef struct
{
	int h;
	int w;
	int l;
} GUTUIE;

bool cmph ( GUTUIE i, GUTUIE j)
{
	return i.l > j.l;
}

bool cmpw ( GUTUIE i, GUTUIE j)
{
	return i.w < j.w;
}

int main (int argc, char **argv)
{
	FILE *fin  = fopen (IN, "r");
	FILE *fout = fopen (OUT, "w");
	int i, aux;
	int N;
	int H, U, current;
	unsigned long G = 0;
	fscanf (fin, "%d%d%d", &N, &H, &U);

	vector <GUTUIE> g, Heap;
	make_heap (Heap.begin(), Heap.end(), cmpw);

	for (i = 0; i < N; ++i)
	{
		GUTUIE a;
		fscanf(fin, "%d%d", &a.h, &a.w);
		if (a.h > H)
		{
			aux = 1;
			continue;
		}
		a.l = (H - a.h)/U + 1;
		g.push_back(a);
	}

	if ( aux == 1)
		N = (int)g.size();

	sort (g.begin(), g.end(), cmph);

	for (i = 0; i < N; ++i)
	{
		if (i == 0)
		{
			current = g[i].l;
			Heap.push_back (g[0]);
			push_heap (Heap.begin(), Heap.end(), cmpw);
			continue;
		}

		if (g[i].l != g[i-1].l)
		{
			while (current > g[i].l)
			{
				current--;
				if (Heap.size())
				{
					G += Heap.front().w;
					pop_heap (Heap.begin(), Heap.end(), cmpw);
					Heap.pop_back();
				}
				else break;
			}
		}
		Heap.push_back (g[i]);
		push_heap (Heap.begin(), Heap.end(), cmpw);
	}

	while (current > 0 && Heap.size())
	{
		G += Heap.front().w;
		pop_heap (Heap.begin(), Heap.end(), cmpw);
		Heap.pop_back();
		current--;
	}

	fprintf (fout, "%lu\n", G);

	fclose (fin);
	fclose (fout);

	return 0;
}