Cod sursa(job #749943)

Utilizator SpiriFlaviuBerbecariu Flaviu SpiriFlaviu Data 19 mai 2012 18:27:52
Problema Algoritmul lui Dijkstra Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.22 kb
#include <fstream>
#include <vector>
#include <set>

#define INF 1000000000

using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

vector< pair<int,int> > v[50001];

bool used[50001];

int c[50001];

int main()
{
    int n,m,x,y,dc;
    fin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>dc;
        v[x].push_back(make_pair(y,dc));
    }
    for(int i=1;i<=n;i++)
        c[i]=INF;
    int curent = 1;
    c[1]=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<v[curent].size();j++)
        {
            if(c[curent]!=INF)
                if(v[curent][j].second + c[curent] < c[v[curent][j].first])
                {
                    c[v[curent][j].first]=v[curent][j].second + c[curent];
                }
        }
        int nmin=1;
        int vmin=INF;
        for(int j=1;j<=n;j++)
            if(!used[j] && c[j]<=vmin)
            {
                nmin=j;
                vmin=c[j];
            }
        used[nmin]=true;
        curent = nmin;

    }

    for(int i=2;i<=n;i++)
        if(c[i]==INF)
            fout<<0<<' ';
        else fout<<c[i]<<' ';
    fout<<'\n';
    fin.close();
    fout.close();
    return 0;
}