Cod sursa(job #1705890)

Utilizator TodeaDariustodea darius TodeaDarius Data 21 mai 2016 01:43:36
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.51 kb
#include<cstdio>
#include<vector>
#include<queue>
#define nmax 1999999999
using namespace std;
struct nod
{
    int inf,ct;
};
vector<nod>v[50010];
int n,m,dmin[50010],nr,ec[50010];
struct ddmin
{
    int vf,l;
};
struct comp
{
    bool operator()(ddmin &a,ddmin &b)
    {
        return a.l<b.l;
    }
};
priority_queue<ddmin,vector<ddmin>,comp>h;
nod makenod(int b,int c)
{
    nod t;
    t.inf=b;
    t.ct=c;
    return t;
}
void citire()
{
    int x,y,z;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        v[x].push_back(makenod(y,z));
    }
    dmin[1]=0;
    for(int i=2;i<=n;i++)
        dmin[i]=nmax;
}
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
 
    citire();
    ddmin x;
    x.vf=1;
    x.l=dmin[1];
    h.push(x);
    while(!h.empty())
    {
        x=h.top();
        ec[x.vf]=0;
        h.pop();
        nr=v[x.vf].size();
        for(int i=0;i<nr;i++)
        {
            if(dmin[x.vf]+v[x.vf][i].ct<dmin[v[x.vf][i].inf])
            {
                dmin[v[x.vf][i].inf]=dmin[x.vf]+v[x.vf][i].ct;
                if(ec[v[x.vf][i].inf]==0)
                {
                    ec[v[x.vf][i].inf]=1;
                    ddmin y;
                    y.vf=v[x.vf][i].inf;
                    y.l=dmin[v[x.vf][i].inf];
                    h.push(y);
                }
            }
        }
    }
    for(int i=2;i<=n;i++)
        if(dmin[i]==nmax)
            printf("0 ");
        else
            printf("%d ",dmin[i]);
    return 0;
}