Cod sursa(job #1282928)

Utilizator span7aRazvan span7a Data 4 decembrie 2014 21:38:52
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 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],n,m,t[100];
struct cmp{
    bool operator()(int a,int b)
    {
        if(cost[a]<cost[b])
            return true;
        else return false;
    }
};
priority_queue<int,vector<int>,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()
{
    int nodcur;
    int i;
    for(i=2;i<=n;i++)
        cost[i]=M;
    heap.push(1);
    while(!heap.empty())
    {
        nodcur=heap.top();
        heap.pop();
        for(i=0;i<v[nodcur].size();i++)
            if(cost[v[nodcur][i].inf]>cost[nodcur]+v[nodcur][i].c)
                {
                    cost[v[nodcur][i].inf]=cost[nodcur]+v[nodcur][i].c;
                       heap.push(v[nodcur][i].inf);
                    t[v[nodcur][i].inf]=nodcur;
                }
    }
    for(i=2;i<=n;i++)
        if(cost[i]==M)
            fprintf(g,"0 ");
        else
            fprintf(g,"%d ",cost[i]);
}
int main()
{
    citire();
    dijkstra();
}