Cod sursa(job #2114779)

Utilizator andysoloAndrei Solo andysolo Data 25 ianuarie 2018 20:54:48
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.37 kb
#include <cstdio>
#include <iostream>
#include <queue>
#include <climits>
#define NMAX 50000+5
#define mp make_pair
#define p pair<int,int>
#define f first
#define s second

using namespace std;

int N,M;
bool c[NMAX];
int D[NMAX];
vector <p> leg[NMAX];

struct comp
{
    bool operator()(int x,int y)
    {
        return D[x]>D[y];
    }
};
priority_queue<int, vector<int>, comp>q;

void dijkstra(int nodi)
{
    for(int i=1; i<=N; i++)
        D[i]=INT_MAX;

    D[nodi]=0;
    q.push(nodi);

    while(!q.empty())
    {
        int nod=q.top();
        q.pop();
        if(c[nod]==true)
            continue;
        c[nod]=true;

        for(int i=0;i<leg[nod].size();i++)
        {
            int vec=leg[nod][i].f;
            int len=leg[nod][i].s;
            if(D[nod]+len<D[vec])
            {
                D[vec]=D[nod]+len;
                q.push(vec);
            }
        }
    }
}

int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    scanf("%d %d",&N,&M);

    for(int i=1; i<=M; i++)
    {
        int x,y,z;
        scanf("%d %d %d",&x,&y,&z);
        leg[x].push_back(mp(y,z));
    }

    dijkstra(1);

    for(int i=2;i<=N;i++)
    {
        if(D[i]==INT_MAX)
            printf("0 ");
        else
        printf("%d ",D[i]);
    }

    return 0;
}