Cod sursa(job #936475)

Utilizator narcis_vsGemene Narcis - Gabriel narcis_vs Data 7 aprilie 2013 13:08:20
Problema Algoritmul lui Dijkstra Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include<fstream>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<queue>
#define In "dijkstra.in"
#define Out "dijkstra.out"
#define Inf 2147483647
#define Nmax 50002
using namespace std;

struct Coord
{
    int nod,cost;
    bool operator <(const Coord &A) const
    {
        return cost>A.cost;
    }
};
bool viz[Nmax];
int sol[Nmax],n;
vector<Coord>graf[Nmax];
priority_queue<Coord>q;

inline void Citire()
{
    int x,y,c,m;
    ifstream fin(In);
    fin>>n>>m;
    Coord p;
    while(m--)
    {
        fin>>x>>y>>c;
        p.nod = y;
        p.cost = c;
        graf[x].push_back(p);
    }
}

inline void Init()
{
    for(int i=2;i<=n;i++)
        sol[i] = Inf;
}

inline void Dijkstra()
{
    int curent,co;
    Coord poz,vec;
    poz.nod = 1;
    poz.cost = 0;
    q.push(poz);
    vector<Coord>::iterator it;
    while(!q.empty())
    {
        poz = q.top();
        q.pop();
        curent = poz.nod;
        viz[curent] = true;
        co = sol[curent];
        for(it=graf[curent].begin();it!=graf[curent].end();it++)
        {
            vec = *it;
            if(viz[vec.nod]) continue;
            if(sol[vec.nod]>co+vec.cost)
            {
                sol[vec.nod] = co+ vec.cost;
                q.push(vec);
            }
        }
    }
}

inline void Afis()
{
    ofstream fout(Out);
    for(int i=2;i<=n;i++)
        fout<<(sol[i]==Inf?0:sol[i])<<" ";
    fout<<"\n";
    fout.close();
}

int main()
{
    Citire();
    Init();
    Dijkstra();
    Afis();
    return 0;
}