Cod sursa(job #406092)

Utilizator jupanubv92Popescu Marius jupanubv92 Data 1 martie 2010 10:20:53
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 2.01 kb
#include<stdio.h>
#include<vector>
#define INF 10000001
#define mp make_pair
#define pb push_back
#define Nmx 50005

using namespace std;

int n,m,viz[Nmx],cost[Nmx],Heap[Nmx*2],nr;
vector< pair<int,int> > G[Nmx];

void schimb(int x,int y)
{
    int aux=Heap[x];
    Heap[x]=Heap[y];
    Heap[y]=aux;
}

void in_jos(int k)
{
    while(k*2<=nr)
    {
        int p=k*2;
        if(p+1<=nr&&cost[Heap[p+1]]<cost[Heap[p]])
            ++p;
        if(cost[Heap[p]]>cost[Heap[k]])
            return;
        viz[Heap[p]]=k;
        viz[Heap[k]]=p;
        schimb(p,k);
        k=p;
    }
}


void in_sus(int k)
{
    while(k>1)
    {
        if(cost[Heap[k/2]]<cost[Heap[k]])
            return;
        viz[Heap[k]]=k/2;
        viz[Heap[k/2]]=k;
        schimb(k,k/2);
        k=k/2;
    }
}

void dijkstra()
{
    vector< pair<int,int> > ::iterator it;
    for(int i=2;i<=n;++i)
    {
        viz[i]=-1;
        cost[i]=INF;
    }
    cost[1]=0;
    Heap[1]=1;
    viz[1]=1;nr=1;
    while(nr)
    {
        int min=Heap[1];
        schimb(1,nr);
        Heap[nr--]=0;
        viz[Heap[1]]=1;
        in_jos(1);
        for(it=G[min].begin();it!=G[min].end();++it)
            if(cost[it->first]>cost[min]+it->second)
            {
                cost[it->first]=cost[min]+it->second;
                if(viz[it->first]==-1)
                {
                    Heap[++nr]=it->first;
                    viz[it->first]=nr;
                    in_sus(nr);
                }
                else in_sus(viz[it->first]);
            }
    }
    for(int i=2;i<=n;++i)
        if(cost[i]==INF)
            printf("0 ");
        else printf("%d ",cost[i]);
}

void citire()
{
    scanf("%d%d",&n,&m);
    int x,y,c;
    for(int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&x,&y,&c);
        G[x].pb(mp(y,c));
    }
}

int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    citire();
    dijkstra();
    return 0;
}