Cod sursa(job #1968362)

Utilizator trutruvasilicaHuhurez Marius trutruvasilica Data 17 aprilie 2017 17:23:59
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
const int Nmax=50010;
struct pi
{
    int x,y;
}ps;
vector<pi>G[Nmax];
vector<pi>::iterator it;
int dist[Nmax];
bitset<Nmax>viz;
struct cmp
{
    bool operator()(const int &a,const int &b)
    {
        return dist[a]>dist[b];
    }
};
priority_queue<int,vector<int>,cmp>Q;
void dijkstra(int start)
{
    dist[start]=0;
    Q.push(start);
    while(Q.size())
    {
        int nod=Q.top();
        Q.pop();
        viz[nod]=0;
        for(it=G[nod].begin();it!=G[nod].end();it++)
        {
            if(dist[it->x]>dist[nod]+it->y)
            {
                dist[it->x]=dist[nod]+it->y;
                if(viz[it->x]==0)
                {
                    viz[it->x]=1;
                    Q.push(it->x);
                }
            }
        }
    }
}
int main()
{
    int n,m,i,a,b,c;
    fin>>n>>m;
    while(m--)
    {
        fin>>a>>ps.x>>ps.y;
        G[a].push_back(ps);
    }
    memset(dist,0x3f3f3f3f,sizeof(dist));
    dijkstra(1);
    for(i=2;i<=n;i++)
    {
        if(dist[i]==0x3f3f3f3f) fout<<"0 ";
        else fout<<dist[i]<<" ";
    }
}