Cod sursa(job #373756)

Utilizator cristikIvan Cristian cristik Data 14 decembrie 2009 23:17:02
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.65 kb
#include <stdio.h>
#define max 100010
const int inf=1<<30;
struct lista
{
    int nod,cost;
    lista *next;
};
lista *g[max],*p;
int n,m,j,d[max],t[max],h[max],poz[max],min,lg,i,c;
void swap(int i,int j)
{
    int t=h[i]; h[i]=h[j]; h[j]=t;
}
void push(int i,int j,int c)
{
    lista *p=new lista;
    p->cost=c;
    p->nod=j;
    p->next=g[i];
    g[i]=p;
}
void push(int x)
{
    while(x/2 && d[h[x]]<d[h[x/2]])
    {
        swap(x,x/2);
        poz[h[x]]=x;
        poz[h[x/2]]=x/2;
        x/=2;
    }
}
void pop(int x)
{
    int y=0;
    while(x!=y)
    {
        y=x;
        if(y*2<=lg && d[h[x]]>d[h[y*2]]) x=y*2;
        if(y*2+1<=lg && d[h[x]]>d[h[y*2+1]]) x=y*2+1;
        swap(x,y);
        poz[h[x]]=x;
        poz[h[y]]=y;
    }
}
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(; m>0; m--)
    {
        scanf("%d%d%d",&i,&j,&c);
        push(i,j,c);//push(j,i,c);
    }
    for(i=2; i<=n; i++) d[i]=inf,poz[i]=-1;
    poz[1]=1; d[1]=0;

    h[++lg]=1;

    while(lg)
    {
        min=h[1];
        swap(1,lg);
        poz[h[1]]=1;
        lg--;
        pop(1);
        for(p=g[min]; p!=NULL; p=p->next)
         if(d[p->nod]>d[min]+p->cost)
         {
             d[p->nod]=d[min]+p->cost;
             if(poz[p->nod]!=-1) push(poz[p->nod]);
             else
             {
                 h[++lg]=p->nod;
                 poz[h[lg]]=lg;
                 push(lg);
             }
         }
    }
    for(i=2; i<=n; i++)
     if(d[i]!=inf) printf("%d ",d[i]);
     else printf("0 ");
    return 0;
}