Cod sursa(job #1936459)

Utilizator GeorgeCalinPetruta George-Calin GeorgeCalin Data 23 martie 2017 09:27:24
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.52 kb
#include <fstream>
#include <climits>
#include <vector>
#include <bitset>
#define NMAX 50002
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

struct NOD{
    int v,c;
};
vector <NOD> graf[NMAX];
int dist[NMAX];
bitset <NMAX> viz;

int main()
{
    int n,m,x,y,c1;
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        dist[i]=INT_MAX;
    }
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>c1;
        NOD nod;
        nod.v=y;
        nod.c=c1;
        graf[x].push_back(nod);
    }
    for(int i=0;i<graf[1].size();i++)
    {
        dist[graf[1][i].v]=graf[1][i].c;
    }
    viz[1]=1;
    int mi=INT_MAX;
    int k;
    for(int i=2;i<=n;i++)
    {
        if(dist[i]<mi)
        {
            mi=dist[i];
            k=i;
        }
    }
    int vi=1;
    while(vi!=0)
    {
        vi=0;
        viz[k]=1;
        for(int i=0;i<graf[k].size();i++)
        {
            int s=graf[k][i].c;
            if(dist[graf[k][i].v]>s+dist[k])
            {
                dist[graf[k][i].v]=s+dist[k];
            }
        }
        int mi=INT_MAX;
        for(int i=1;i<=n;i++)
        {
            if(viz[i]==0&&mi>dist[i])
            {
                vi=1;
                mi=dist[i];
                k=i;
            }
        }
    }
    for(int i=2;i<=n;i++)
    {
        if(dist[i]!=INT_MAX)
        {
            fout<<dist[i]<<" ";
        }
        else
        {
            fout<<-1<<" ";
        }
    }
    return 0;
}