Cod sursa(job #439523)

Utilizator ctvalentinMarcu Valentin ctvalentin Data 11 aprilie 2010 16:48:04
Problema Gutui Scor 10
Compilator c Status done
Runda teme_upb Marime 2.52 kb
#include<stdlib.h>
#include<stdio.h>
//int max[];
int N=0, H=0, U=1, nr=0, i, ii, j, k=0, aux=0, r, t, a=0, b=0, suma=0, **max, *diag, *ord;
FILE *f, *g;

int compare_int (const void *Y, const void *Z)
{
       int y = *((int *)Y);
       int z = *((int *)Z);
       if (y > z)
       {
               return -1;
       }
       else
       {
               if (y < z)
               {
                       return 1;
               }
               else
               {
                       return 0;
               }
       }
}

int main()
{
    f = fopen("gutui.in","r");
    if(!f) {printf("eroare f"); return 1;}
    g = fopen("gutui.out","w");
    if(!f) {printf("eroare g"); return 2;}
    fscanf(f,"%i", &N);
    fscanf(f,"%i", &H);
    fscanf(f,"%i", &U);
    nr = H/U;            //nr=numarul de nivele si totodata numarul maxim de gutui ce pot fi culese din pom
    max = calloc ((nr+2),sizeof(int*));
    ord = calloc ((nr+2),sizeof(int*));
    for(i=0;i<=nr+1;i++) { max[i] = calloc (N+nr+1,sizeof(int));
                        }
    //citirea datelor 
    for(i=0;i<N;i++)
    {
          
     
          fscanf(f,"%i", &a);  //a este inaltime initiala
          fscanf(f,"%i", &b);  //b este greutatea
          if(a<=H)
          {
            j= nr - (a-1)/U;          //j indica ordinea de evaluare, adica nivelul pe care se va gasi gutuia citita
         
          ord[j]++; r = ord[j];  //ord[j]= numarul de gutui de pe linia j in matricea max

          max[j][r]=b;  //retin greutatea
          }
    }
    
    for(i=1; i<=nr; i++)  qsort((void *)(max[i]), N, sizeof(int), compare_int);  //sortez fiecare linie descrescator dupa greutate; O(nr*logN)
    
    diag = calloc (N+nr+1,sizeof(int));
    
    for(i = 0; i <= N+nr; i++) diag[i]=0;
    k=0;
    for(i=1; i<=nr; i++)
        { for(j=0; j<=i-1; j++)  //de pe nivelul i pot culege maximum i gutui
           { 
               if(max[i][j] > 0) { diag[k] = max[i][j];  k++; }  //pun in diag doar gutuile, nu si locurile libere
               else j = i-1;  //altfel intrerup for-ul interior pentru i-ul curent
           }
        } 
    qsort((void *)diag, N+nr+1, sizeof(int), compare_int);  //sortez descrescator dupa greutate gutuile din vectorul diag
  
    for(j=0; j<nr; j++) suma+=diag[j];  //pot lua maximum nr gutui (daca in diag am pus mai putin de nr gutui, restul elementelor vor fi 0 si nu vor influenta suma
    fprintf(g,"%i", suma);
  
    free (max);
    fclose(f);
    fclose(g);
 
    return 0;
}