Cod sursa(job #1782918)

Utilizator andreeainfo_dAndreea Dutulescu andreeainfo_d Data 18 octombrie 2016 17:12:25
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.98 kb
#include <bits/stdc++.h>
using namespace std;
int dist[50005],i,l,n,m,x,y,xy,nod,val;
vector < pair <int,int> >v[50005];
#define INF (1<<30)
struct andreea
{
    int value,node;
}aint[200005];
void build (int nod, int st, int dr)
{
    int mij;
    if(st==dr)
    {
        aint[nod].value=dist[st];
        aint[nod].node=st;
        return;
    }
    mij=(st+dr)/2;
    build(nod*2,st,mij);
       build(nod*2+1,mij+1,dr);
    if(aint[nod*2].value<aint[nod*2+1].value)
    {
        aint[nod].value=aint[nod*2].value;
        aint[nod].node=aint[nod*2].node;
    }
    else
    {
        aint[nod].value=aint[nod*2+1].value;
        aint[nod].node=aint[nod*2+1].node;
    }
}
void update(int nod, int st, int dr, int poz, int val)
{
    int mij;
    if(st==dr)
    {
        aint[nod].value=val;
        return;
    }
    mij=(st+dr)/2;
    if(poz<=mij)update(nod*2,st,mij,poz,val);
    else update(nod*2+1,mij+1,dr,poz,val);
    if(aint[nod*2].value<aint[nod*2+1].value)
    {
        aint[nod].value=aint[nod*2].value;
        aint[nod].node=aint[nod*2].node;
    }
    else
    {
        aint[nod].value=aint[nod*2+1].value;
        aint[nod].node=aint[nod*2+1].node;
    }
}
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%d%d%d",&x,&y,&xy);
        v[x].push_back(make_pair(y,xy));
        dist[i]=INF;
    }
    dist[1]=0;
    build(1,1,n);
  //  return 0;
    while(aint[1].value!=INF)
    {
        nod=aint[1].node;
        val=aint[1].value;
        update(1,1,n,nod,INF);
        for(i=0;i<v[nod].size();i++)
        {
            if(v[nod][i].second+val<dist[v[nod][i].first])
            {
                dist[v[nod][i].first]=v[nod][i].second+val;
                update(1,1,n,v[nod][i].first,dist[v[nod][i].first]);
            }
        }
    }
    for(i=2;i<=n;i++)
        printf("%d ",dist[i]);
    return 0;
}