Cod sursa(job #2661872)

Utilizator alexia208160Popescu Alexia Maria alexia208160 Data 22 octombrie 2020 20:21:07
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.38 kb
#include <fstream>
#include <queue>

using namespace std;

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

struct Road
{
    int len, connect;
    bool operator < (const Road other) const
    {
        Road bleah;
        return bleah.len < other.len;
    }
};

priority_queue <Road> pq;

bool visit[50005];
int final[50005];
vector <Road> v[50005];

void Dijkstra()
{
    while(!pq.empty())
    {
        Road start = pq.top();
        int node = start.connect;
        pq.pop();
        visit[node] = 0;
        for(int i = 0; i < v[node].size(); i++)
        {
            int length = v[node][i].len + final[node];
            if(length < final[v[node][i].connect] || (final[v[node][i].connect] == 0 && v[node][i].connect != 1))
            {
                final[v[node][i].connect] = length;
                if(visit[v[node][i].connect] == 0)
                {
                    pq.push({length, v[node][i].connect});
                    visit[v[node][i].connect] = 1;
                }
            }
        }
    }

}

int main(void)
{
    int n, m;
    fin >> n >> m;
    for(int i = 0; i < m; i++)
    {
        int a;
        Road node;
        fin >> a >> node.connect >> node.len;
        v[a].push_back(node);
    }
    pq.push({0, 1});
    Dijkstra();
    for(int i = 2; i <= n; i++)
        fout << final[i] <<' ';
}