Cod sursa(job #1638718)

Utilizator redducks100Andronache Simone redducks100 Data 8 martie 2016 08:48:10
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.35 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

#define NMAX 21000
int n,m,graph[NMAX][NMAX];

int minDistance(int dist[NMAX],bool fol[NMAX])
{
    int minn = INFINITY,minIndex=0;

    for(int i=0;i<n;i++)
    {
        if(fol[i]==false && dist[i]<=minn)
        {
            minn = dist[i];
            minIndex = i;
        }
    }

    return minIndex;
}

void print(int dist[NMAX])
{
    for(int i=1;i<n;i++)
    {
        if(dist[i]==INFINITY)
            g<<0<<" ";
        else
            g<<dist[i]<<" ";
    }
}

void dijkstra(int source)
{
    int dist[NMAX];
    bool fol[NMAX];

    for(int i=0;i<n;i++)
    {
        dist[i] = INFINITY;
        fol[i] = false;
    }

    dist[source] = 0;

    for(int i=0;i<n-1;i++)
    {
        int u = minDistance(dist,fol);

        fol[i] = true;

        for(int j=0;j<n;j++)
        {
            if(!fol[j] && graph[u][j] && dist[u] != INFINITY && dist[u] + graph[u][j] < dist[j])
            {
                dist[j] = dist[u] + graph[u][j];
            }
        }
    }

    print(dist);
}

int main()
{
    f>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int x,y,cost;
        f>>x>>y>>cost;
        graph[x-1][y-1]=graph[y-1][x-1]=cost;
    }
    dijkstra(0);

    return 0;
}