Cod sursa(job #442140)

Utilizator emilia.ciobanuCiobanu Emilia Maria Silvia emilia.ciobanu Data 13 aprilie 2010 22:02:15
Problema Gutui Scor 100
Compilator cpp Status done
Runda teme_upb Marime 1.72 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, GUTUIE);

bool cmpw (GUTUIE, GUTUIE);

int main (int argc, char **argv)
{
	FILE *fin  = fopen (IN, "r");
	FILE *fout = fopen (OUT, "w");
	vector <GUTUIE> g, Heap;
	unsigned long G = 0;
	int i;
	int N;
	int H, U, current;
	char aux;

	fscanf (fin, "%d%d%d", &N, &H, &U);
	make_heap (Heap.begin(), Heap.end(), cmpw);

	/* Citire selectiva, in cazul in care o parte din gutui sunt
	 * la o inaltime > H
	 */
	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);
	}

	/* Daca am gasit astfel de gutui, nu am nevoie sa lucrez cu ele => actualizez N */
	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;
}

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

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