Cod sursa(job #463396)

Utilizator simona.poilincaSimona Poilinca simona.poilinca Data 15 iunie 2010 17:08:54
Problema Gutui Scor 80
Compilator c Status done
Runda teme_upb Marime 2.58 kb
// tema 1 pa ex.2
#include <stdio.h>
#include <stdlib.h>

int N, H, U;

  // mergesort ce sorteaza descrescator vectorul x
  // vectorii x si w tin locul unei structuri de elemente ce leaga x[i] cu w[i]
void merge(int *x, int *w, int l, int m, int r) {
	int i = l, j = m+1, k = 0;
	int *y, *z;
	y = (int *)malloc((r-l+1) * sizeof(int));
	z = (int *)malloc((r-l+1) * sizeof(int));
	while(i<=m && j<=r) {
		if(x[i] >= x[j])
		{
			y[k] = x[i];
			z[k++] = w[i++];
		}
		else
		{
			y[k] = x[j];
			z[k++] = w[j++];
		}
	}

	while(i<=m)
	{
		y[k] = x[i];
		z[k++] = w[i++];
	}
	while(j<=r)
	{
		y[k] = x[j];
		z[k++] = w[j++];
	}
	k = 0;
	for(i = l; i<=r; i++)
	{
		x[i] = y[k];
		w[i] = z[k++];
	}
	free(y);
	free(z);
}

void mergesort(int *x, int *w, int l, int r) {
	int m;
	if(l < r) {
		m = (l + r) / 2;
		mergesort(x, w, l, m);
		mergesort(x, w, m + 1, r);
		merge(x, w, l, m, r);
	}
}

int main() {
	FILE *fin = fopen("gutui.in", "r");
	FILE *fout = fopen("gutui.out", "w");

	fscanf(fin, "%d%d%d", &N, &H, &U);
	int i, j, k, nrGutuiMax, cantar = 0, min;
	
  // initializarea a 3 vectori:
  // h - vectorul inaltimilor la care se afla gutuile
  // g - vectorul greutatilor gutuilor
  // ajunsa - vector ce retine la ce moment o gutuie poate fi ajunsa
	int *h, *g, *ajunsa;

	h = (int *)malloc(N * sizeof(int));
	g = (int *)malloc(N * sizeof(int));
	ajunsa = (int *)malloc(N * sizeof(int));

  // citirea datelor pentru fiecare gutuie i
	for (i = 0; i < N; i++)
	{
		fscanf(fin, "%d%d", &h[i], &g[i]);
	}

  // sortare descrescatoare dupa inaltime
	mergesort(h, g, 0, N - 1);
	
  // memorarea inaltimii celei mai de jos gutui
	min = h[N - 1];

  // sortarea descrescatoare dupa greutatea gutuilor
	mergesort(g, h, 0, N - 1);

  // desemnarea momentului pana la care o gutuie inca poate fi ajunsa
	for (i = 0; i < N; i++)
	{
		ajunsa[i] = (H - h[i] + U)/U;
	}

  /* memorarea numarului maxim de gutui ce pot fi culse = cat timp cea mai de */
  /* jos gutuie inca poate fi culeasa */
	nrGutuiMax = (H - min + U)/U;
	
	i = 0;
	j = 0;

  /* functia de culegere a gutuilor */
  /* cat timp nu am luat in considerare toate gutuile, sau nu am cules */
  /* maximul de gutui, daca o gutuie inca poate fi ajunsa, este culeasa */
  /* si se reduc momentele pentru toate gutuile aflate pe aceeasi creanga */
  /* sau pe crengile mai apropiate de pamant */
	while (i < N && j < nrGutuiMax)
	{
		if (ajunsa[i] > 0)
		{
			cantar += g[i];
			j++;
			for (k = i+1 ; k < N; k++)
				if (ajunsa[k] >= ajunsa[i])
					ajunsa[k]--;
		}
		
		i++;
	}

	fprintf(fout, "%d", cantar);

	free(h);
	free(g);
	free(ajunsa);

	fclose(fin);
	fclose(fout);
	return 0;
}