Cod sursa(job #936456)

Utilizator narcis_vsGemene Narcis - Gabriel narcis_vs Data 7 aprilie 2013 12:00:47
Problema Algoritmul lui Dijkstra Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.82 kb
#include<fstream>
#include<vector>
#include<set>
#include<bitset>
#include<cstdlib>
#include<cstring>
#define In "dijkstra.in"
#define Out "dijkstra.out"
#define Inf 2147483647
#define Nmax 50002
using namespace std;
int sol[Nmax],n;
vector<pair<int,int> >graf[Nmax];
set< pair<int,int> >S;
bitset<Nmax>viz;
inline void Citire()
{
    int x,y,c,m;
    ifstream fin(In);
    char s[100],*p;
    fin.getline(s,100);
    p = strtok(s," ");
    n = atoi(p);
    p = strtok(NULL," ");
    m = atoi(p);
    while(m--)
    {
        fin.getline(s,100);
        p = strtok(s," ");
        x = atoi(p);
        p = strtok(NULL," ");
        y = atoi(p);
        p = strtok(NULL," ");
        c = atoi(p);
        graf[x].push_back(make_pair(c,y));
    }
}

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

inline void Dijkstra()
{
    pair<int,int> poz,vec;
    S.insert(make_pair(0,1));
    vector< pair<int,int> >::iterator it;
    set< pair<int,int> >::iterator its;
    while(!S.empty())
    {
        its = S.begin();
        poz = *its;
        viz[poz.second] = true;
        S.erase(its);
        for(it=graf[poz.second].begin();it!=graf[poz.second].end();it++)
        {
            vec = *it;
            if(viz[vec.second]==true)
                continue ;
            if(sol[vec.second]>sol[poz.second]+vec.first)
            {
                sol[vec.second] = sol[poz.second] + vec.first;
                S.insert(vec);
            }
        }
    }
}

inline void Afis()
{
    ofstream fout(Out);
    for(int i=2;i<=n;i++)
    {
        if(sol[i]==Inf)
            fout<<"0 ";
        else
            fout<<sol[i]<<" ";
    }
    fout<<"\n";
    fout.close();
}
int main()
{
    Citire();
    Init();
    Dijkstra();
    Afis();
    return 0;
}