Cod sursa(job #1506679)

Utilizator trutruvasilicaHuhurez Marius trutruvasilica Data 20 octombrie 2015 21:16:05
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.23 kb
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#include <cstring>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
const int N=50001,inf=0x3f3f3f3f;
struct nod
{
    int x,cost;
};
vector<nod>H[N];
vector<nod>::iterator it;
int dist[N];
struct comp
{
    bool operator() (const nod &A,const nod &B)
    {
        return A.cost>B.cost;
    }
};
priority_queue <nod, vector<nod> , comp> Q;
int main()
{
    int n,m,a,i;
    nod A,B;
    fin>>n>>m;
    memset(dist,inf,sizeof(dist));
    for(i=1;i<=n;i++)
    {
        fin>>a>>A.x>>A.cost;
        H[a].push_back(A);
    }
    for(it=H[1].begin();it!=H[1].end();it++)
    {
        dist[it->x]=it->cost;
        Q.push(*it);
    }
    while(Q.size())
    {
        A=Q.top();
        for(it=H[A.x].begin();it!=H[A.x].end();it++)
        {
            if(dist[it->x]>dist[A.x]+it->cost)
            {
                dist[it->x]=dist[A.x]+it->cost;
                B.x=it->x;
                B.cost=dist[B.x];
                Q.push(B);
            }
        }
        Q.pop();
    }
    for(i=2;i<=n;i++)
    {
        if(dist[i]==inf) fout<<"0 ";
        else fout<<dist[i]<<" ";
    }
}