Cod sursa(job #440450)

Utilizator DetudaArthur Koestler Detuda Data 12 aprilie 2010 01:15:55
Problema Gutui Scor 60
Compilator cpp Status done
Runda teme_upb Marime 2.47 kb
#include <stdio.h>
#include <malloc.h>
#include <algorithm>
#include <vector>
using namespace std;


#define u_int unsigned int


void merge(u_int **a, int low, int high, int mid){
	int i, j, k;
	u_int **c;
	
	c = (u_int**)malloc((high+1)*sizeof(u_int*));
	for(i = 0; i < high+1; i ++)
		c[i] = (u_int*)malloc(2*sizeof(u_int));
		
	i = low;
	j = mid+1;
	k = low;
	
	while((i <= mid) && (j <= high)){
		if(a[i][0] > a[j][0]){
			c[k][0] = a[i][0];
			c[k][1] = a[i][1];
			i ++;
		}
		else{
			c[k][0] = a[j][0];
			c[k][1] = a[j][1];
			j ++;
		}
		k ++;
	}
	while(i <= mid){
		c[k][0] = a[i][0];
		c[k][1] = a[i][1];
		k ++;
		i ++;
	}
	
	while(j <= high){
		c[k][0] = a[j][0];
		c[k][1] = a[j][1];
		k ++; 
		j ++;
	}
	
	for(i = low; i < k; i ++){
		a[i][0] = c[i][0];
		a[i][1] = c[i][1];
	}
	
	for(i = 0; i < high+1; i ++)
		free(c[i]);
	free(c);
	
	return;
}

void merge_sort(u_int **a, int low, int high){
	int mid;
	if(low<high){
		mid = (low+high)/2;
		merge_sort(a, low, mid);
		merge_sort(a, mid+1, high);
		merge(a, low, high, mid);
	}
	return;
}

int main(){
	FILE *f, *g;
	f = fopen ("gutui.in", "r");
	g = fopen ("gutui.out", "w");
	u_int H, U, **a, cules = 0, h_mom;
	int N, i, n, p = 0;
	vector<u_int> momentan;
	vector<u_int>::iterator it;
	
	fscanf(f, "%d %u %u", &N, &H, &U);
	n = N;
	a = (u_int**)malloc(N*sizeof(u_int*));
	for(i = 0; i < N; i ++){
		a[i] = (u_int*)malloc(2*sizeof(u_int));
		fscanf(f, "%u %u", &a[i][0], &a[i][1]);
	}

//merge_sort descrescator		
	merge_sort(a, 0, N-1);
	
	/*for(i = 0; i < N; i ++)
		fprintf(g, "%u %u\n", a[i][0], a[i][1]);
	*/
	
//pornesc de la ultimul nivel	
	if(H%U == 0)
		h_mom = U;
	else
		h_mom = H - ((H / U) * U);
	make_heap(momentan.begin(), momentan.end());
//incep sa "culeg" gutui, pornind de la ultimul nivel
	while((h_mom <= H) && (N > 0 || momentan.size() > 0)){
		if(momentan.size() > 0){
			cules = cules + momentan.front();
			pop_heap(momentan.begin(), momentan.end());
			momentan.pop_back();
			h_mom = h_mom + U;
		}
		else
			if(p)
				h_mom = h_mom + U;
		p = 1;
//parcurge un nivel
		while(N > 0){
//s-a terminat nivelul -> iese din while
			if(a[N-1][0] > h_mom)
				break;
//pune gutuile de pe nivel in momentan			
			momentan.push_back(a[N-1][1]);
			N --;
			push_heap(momentan.begin(), momentan.end());

		}
	}
	
	for(i = 0; i < n; i ++)
		free(a[i]);
	free(a);
	
	fprintf(g, "%u", cules);
	
	return 0;
}