Cod sursa(job #1282971)

Utilizator span7aRazvan span7a Data 4 decembrie 2014 22:04:27
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.37 kb
#include<cstdio>
#include<vector>
#include<queue>
#define M 1<<30
using namespace std;
FILE *f=fopen("dijkstra.in","r");
FILE *g=fopen("dijkstra.out","w");
struct nod{
    int inf,c;
};
vector<nod>v[50001];
int cost[50001],viz[50001],n,m;
struct cmp{
    bool operator()(nod a,nod b)
    {
        if(cost[a.inf]>cost[b.inf])
            return true;
        else return false;
    }
};
priority_queue<nod,vector<nod>,cmp>heap;
void citire()
{
    int i,x,y,z;
    nod a;
    fscanf(f,"%d%d",&n,&m);
    for(i=1;i<=m;i++)
    {
        fscanf(f,"%d%d%d",&x,&y,&z);
        a.inf=y;
        a.c=z;
        v[x].push_back(a);
    }
}
void dijkstra()
{
    nod a,nodcur;
    int i;
    for(i=2;i<=n;i++)
        cost[i]=M;
    a.inf=1;a.c=0;heap.push(a);
    while(!heap.empty())
    {
        nodcur=heap.top();
        heap.pop();
        //viz[nodcur.inf]=1;
        for(i=0;i<v[nodcur.inf].size();i++)
            if(cost[v[nodcur.inf][i].inf]>cost[nodcur.inf]+v[nodcur.inf][i].c)
                {
                    cost[v[nodcur.inf][i].inf]=cost[nodcur.inf]+v[nodcur.inf][i].c;
                       heap.push(v[nodcur.inf][i]);

                }
    }
    for(i=2;i<=n;i++)
        if(cost[i]==M)
            fprintf(g,"0 ");
        else
            fprintf(g,"%d ",cost[i]);
}
int main()
{
    citire();
    dijkstra();
}