Cod sursa(job #2262002)

Utilizator Dragne.Andrei11Dragne Andrei Dragne.Andrei11 Data 16 octombrie 2018 21:16:26
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.69 kb
#include <bits/stdc++.h>

using namespace std;

const int nmax=50005;

vector <pair <int, int> > graph_orientat[nmax];
vector <pair <int, int> > coada;
int cost[nmax];
bool visited[nmax];

void dijkstra(int start_node)
{
    cost[start_node]=0;
    visited[start_node]=true;
    for(int i=0;i<graph_orientat[start_node].size();i++)
        coada.push_back(make_pair(graph_orientat[start_node][i].first+cost[start_node], graph_orientat[start_node][i].second));
    make_heap(coada.begin(), coada.end());
    while(!coada.empty())
    {
        int ll;
        ll=coada.size();
        int current_node=coada.back().second;
        if(visited[current_node]==false)
        {
            visited[current_node]=true;
            cost[current_node]=coada.back().first;
            for(int i=0;i<graph_orientat[current_node].size();i++)
                coada.push_back(make_pair(graph_orientat[current_node][i].first+cost[current_node], graph_orientat[current_node][i].second));
            coada[ll-1].first=2*nmax;
            make_heap(coada.begin(), coada.end());
            sort_heap(coada.begin(), coada.end());
            coada.pop_back();
            make_heap(coada.begin(), coada.end());
        }
        else
        {
            coada.pop_back();
        }
    }
}

int main()
{
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);
    int n, m;

    scanf("%d%d", &n, &m);
    for(int i=1;i<=m;i++)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        graph_orientat[a].push_back(make_pair(c, b));
    }
    dijkstra(1);
    for(int i=2;i<=n;i++)
        printf("%d ", cost[i]);
    printf("\n");

    return 0;
}