Pagini recente » Cod sursa (job #220840) | Cod sursa (job #1238674) | Cod sursa (job #141670) | Cod sursa (job #2319462) | Cod sursa (job #440450)
Cod sursa(job #440450)
#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;
}