Cod sursa(job #2551757)

Utilizator AlbertUngureanuAlbert AlbertUngureanu Data 20 februarie 2020 10:04:08
Problema Algoritmul lui Dijkstra Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

int N,M;
int Cost[50001];
bool Z[50001],Viz[50001];
vector< pair<int,int> >Ad[50001];

void citire()
{
    int x,y,z;
    fin>>N>>M;
    for(int i=1;i<=M;i++)
    {
        fin>>x>>y>>z;
        Ad[x].push_back({y,z});
        if(x==1)
        {
            Cost[y]=z;
            if(z==0)
                Z[y]=1;
        }
    }
}

void dijkstra(int x)
{
    deque<int>Q;
    int P;
    Viz[1]=1;
    Q.push_front(1);
    while(!Q.empty())
    {
        P=Q.front();
        Q.pop_front();
        for(unsigned i=0;i<Ad[P].size();i++)
            if((Cost[Ad[P][i].first]==0 && !Z[Ad[P][i].first]) || Cost[Ad[P][i].first]>Cost[P]+Ad[P][i].second)
            {
                Cost[Ad[P][i].first]=Cost[P]+Ad[P][i].second;
                Q.push_front(Ad[P][i].first);
                Viz[Ad[P][i].first]=1;
            }
            else
                if(!Viz[Ad[P][i].first] && Cost[Ad[P][i].first]>Cost[P])
                    Q.push_back(Ad[P][i].first);
    }
}

void afisare()
{
    for(int i=2;i<=N;i++)
        fout<<Cost[i]<<" ";
}

int main()
{
    citire();
    dijkstra(1);
    afisare();
    return 0;
}