Cod sursa(job #803456)

Utilizator TeodoraTanaseTeodora Tanase TeodoraTanase Data 27 octombrie 2012 17:04:20
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.41 kb
#include <cstdio>
#include <vector>
#include <queue>

#define NMAX 50001
#define INF 999999999

using namespace std;

FILE *inFile = fopen ("dijkstra.in", "r");
FILE *outFile = fopen ("dijkstra.out", "w");

struct nod
{
    int v;
    int cost;
};

int n;
int cost[NMAX];

struct comp
{
    bool operator () (int a, int b)
    {
        return cost[a] < cost[b];
    }
};

vector <nod> G[NMAX];
priority_queue <int, vector <int>, comp> Q;

void read()
{
    int m;
    int x;
    nod y;

    fscanf (inFile, "%d %d\n", &n, &m);
    while (m --)
    {
        fscanf (inFile, "%d %d %d\n", &x, &y.v, &y.cost);
        G[x].push_back(y);
    }
}

void dijkstra()
{
    for (int i = 2; i <= n; cost[i] = INF, ++ i);

    Q.push(1);
    while (!Q.empty())
    {
        int v = Q.top();
        Q.pop();

        for (unsigned int i = 0; i < G[v].size(); ++ i)
        {
            nod aux = G[v][i];

            if (cost[v] + aux.cost < cost[aux.v])
            {
                cost[aux.v] = cost[v] + aux.cost;
                Q.push(aux.v);
            }
        }
    }
}

void write()
{
    for (int i = 2; i <= n; ++ i)
        if (cost[i] == INF)
            fprintf (outFile, "0 ");
        else
            fprintf (outFile, "%d ", cost[i]);

    fprintf (outFile, "\n");
}

int main()
{
    read();
    dijkstra();
    write();

    return 0;
}