Cod sursa(job #1327332)

Utilizator andreiulianAndrei andreiulian Data 26 ianuarie 2015 17:05:01
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include<fstream>
#include<set>
#include<vector>
using namespace std;
int N,M,r[50005],contor;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
set< pair<int,int> > s;
vector <int> v[50005],d[50005];
bool viz[50005];
int main()
{
    in>>N>>M;
    int i,a,b,c,min;
    for(i=1;i<=N;++i) v[i].push_back(0),d[i].push_back(0),r[i]=1<<30;
    r[1]=0;
    for(i=1;i<=M;++i)
    {
        in>>a>>b>>c;
        v[a].push_back(b);
        v[a][0]++;
        d[a].push_back(c);
        if(c<min) min=c;
    }
    s.insert(make_pair(0,1));
    int dd,nn;
    while(s.size())
    {
        dd=s.begin()->first;
        nn=s.begin()->second;
        if(viz[nn]) s.erase(s.begin());
        else
        {
            viz[nn]=1;
            contor++;
            for(i=1;i<=v[nn][0];++i)
            {
                if(r[ v[nn][i] ] > r[nn] + d[nn][i] )
                {
                    r[ v[nn][i] ]= r[nn] + d[nn][i];
                    s.insert( make_pair(r[ v[nn][i] ],v[nn][i]) );

                }
            }
        }
    }
    for(i=2;i<=N;++i) out<<r[i]<<' ';
    out<<'\n';
}