Cod sursa(job #345391)

Utilizator MarcvsHdrMihai Leonte MarcvsHdr Data 2 septembrie 2009 20:35:21
Problema PScNv Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.2 kb
# include <stdio.h>
# include <stdlib.h>
# include <string.h>

//7:09

const long int MAXN=250000;
const long int MAXM=500000;

typedef struct MUCHIE
        {
        long int a,b,k;
        };

long int used[MAXN+1],pozh[MAXN+1],heap[MAXM+1],raza[MAXN+1],heaplen,n,m,source,sink;
MUCHIE muchie[MAXM+1];

typedef struct NOD
        {
        long int val;
        NOD *next;
        };

typedef struct LISTA
        {
        NOD *begin,*end;
        };

LISTA ladiac[MAXN+1];

void add(LISTA &l, long int val)
{
     NOD *p=(NOD*) malloc(sizeof(NOD));
     p->val=val;
     p->next=NULL;
     if (l.begin)
        {
        l.end->next=p;
        l.end=p;
        }
     else
         {
         l.begin=p;
         l.end=p;
         }
}

void float_heap(long int i)
{
     long int aux;
     while (i/2 && raza[heap[i]]<raza[heap[i/2]])
           {
           aux=heap[i];
           heap[i]=heap[i/2];
           heap[i/2]=aux;
           pozh[heap[i]]=i;
           pozh[heap[i/2]]=i/2;
           i/=2;
           }
}

void submerge_heap(long int i)
{
     long int min,aux;
     do
       {
       min=i;
       if (2*i  <=heaplen && raza[heap[2*i  ]]<raza[heap[min]]) min=2*i;
       if (2*i+1<=heaplen && raza[heap[2*i+1]]<raza[heap[min]]) min=2*i+1;
       if (min==i) break;
       aux=heap[i];
       heap[i]=heap[min];
       heap[min]=aux;
       pozh[heap[i]]=i;
       pozh[heap[min]]=min;
       i=min;
       } while  (1);
}

void insert_heap(long int val)
{
     heap[++heaplen]=val;
     pozh[heap[heaplen]]=heaplen;
     float_heap(heaplen);
}

long int extract_heap()
{
     long int sol=heap[1];
     heap[1]=heap[heaplen];
     heaplen--;
     submerge_heap(1);
     return sol;
}

long int max(long int a, long int b) {if (a>b) return a; return b;}

void load(long int nod)
{
     NOD *p=ladiac[nod].begin;
     while (p)
           {
           if (!used[muchie[p->val].b])
           if (!pozh[muchie[p->val].b]) 
              {
              raza[muchie[p->val].b]=max(raza[nod],muchie[p->val].k);
              insert_heap(muchie[p->val].b);
              }
           else if (raza[muchie[p->val].b]>max(raza[nod],muchie[p->val].k))
                {
                raza[muchie[p->val].b]=max(raza[nod],muchie[p->val].k);
                float_heap(pozh[muchie[p->val].b]);
                }
           p=p->next;
           }
}

void citire()
{
     char buff[1000],*p;
     FILE *f=fopen("pscnv.in","r");
     fscanf(f,"%ld%ld%ld%ld",&n,&m,&source,&sink);
     fgets(buff,1000,f);
     long int i;
     for (i=1;i<=m;i++)
         {
         fscanf(f,"%ld%ld%ld",&muchie[i].a,&muchie[i].b,&muchie[i].k);
         add(ladiac[muchie[i].a],i);
         }
     fclose(f);
}

void scrie(long int sol)
{
     FILE *g=fopen("pscnv.out","w");
     fprintf(g,"%ld\n",sol);
     fclose(g);
}

long int calculeaza()
{
     long int m;
     used[source]=1;
     load(source);
     while (!used[sink])
           {
           m=extract_heap();
           used[m]=1;
           load(m);
           }
     return raza[sink];
}

int main()
{
    citire();
    scrie(calculeaza());
    return 0;
}