Cod sursa(job #1566549)

Utilizator tc_iuresiures tudor-cristian tc_iures Data 12 ianuarie 2016 11:50:46
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.24 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

const int Nmax = 50005;
const int INF  = 2000000000;

vector<pair<int,int> > G[Nmax];
int N, M, D[Nmax];

void read()
{
    ifstream f("dijkstra.in");
    f >> N >> M;
    for(int i = 0; i < M; i ++)
    {
        int x, y, z;
        f >> x >> y >> z;
        G[x].push_back(make_pair(y,z));
    }
    f.close();
}

void solve(int k)
{
    for(int i = 1; i <= N; i ++)
    {
        D[i] = INF;
    }
    D[k] = 0;
    queue<int> Q;
    Q.push(k);
    while(!Q.empty())
    {
        int Node = Q.front();
        Q.pop();
        for(int i  = 0; i < G[Node].size(); i ++)
        {
            int Ngh = G[Node][i].first;
            if(D[Ngh] > D[Node] + G[Node][i].second)
            {
                D[Ngh] = D[Node] + G[Node][i].second;
                Q.push(Ngh);
            }
        }
    }
}

void print()
{
    ofstream g("dijkstra.out");
   for(int i = 2; i <= N; i ++)
   {
      if(D[i] == INF)
      {
         g << 0 << " ";
      }
      else
      {
         g << D[i] << " ";
      }
   }

    g.close();
}

int main()
{
    read();
    solve(1);
    print();
    return 0;
}