Cod sursa(job #1034335)

Utilizator ionicaion ionica Data 17 noiembrie 2013 19:36:45
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.45 kb
#include<fstream>
#include<vector>
#include<queue>
#include<utility>
#define NMAX 50001
#define INF 2500010
#include<stdio.h>
using namespace std;
ifstream f("dijkstra.in");
//FILE *f=fopen("dijkstra.in","r");
ofstream g("dijkstra.out");
//FILE *g=fopen("dijkstra.out","w");
int n, m, D[NMAX];
vector<pair<int, int> > B[NMAX];
priority_queue <pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > H;
void Citeste()
{
    int i, x, y, c;
    f>>n>>m;
    //fscanf(f,"%d %d",&n,&m);
    for (i=1; i<=m; ++i)
    {
        f>>x>>y>>c;
        //fscanf(f,"%d %d %d",&x,&y,&c);
        B[x].push_back(make_pair(c, y));
    }
     D[1]=0;
    for (i=2; i<=n; ++i) D[i]=INF;
}
void Dijkstra()
{
    pair<int, int> pr;
    int z,lg;

    H.push(make_pair(0, 1));
    while (!H.empty())
    {
        int i,j;
        pr=H.top(); H.pop();
        i=pr.second;
        for(j=0;j<B[i].size();j++)
            {
                z=B[i][j].second;;
                lg=D[i]+B[i][j].first;
                if(lg<D[z])
                {
                    H.push(make_pair(lg,z));
                    D[z]=lg;
                }
            }
     }
}
void Scrie()
{
    int i;
    for (i=2; i<=n; ++i)
        if (D[i]==INF) g<<"0 ";
             // fprintf(g,"%d ",0);
            else g<<D[i]<<" ";
               //fprintf(g,"%d ",D[i]);
}
int main()
{
    Citeste();
    Dijkstra();
    Scrie();
   return 0;
}