Cod sursa(job #2830057)

Utilizator TeodoraMaria123Serban Teodora Maria TeodoraMaria123 Data 9 ianuarie 2022 14:03:39
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.79 kb
#include <fstream>
#include <vector>
#include <bitset>
#include <queue>
using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
const int nmax=5e4;
vector <pair <int,int> > v[nmax+5];
vector <int> dist;
bitset <50005> viz;
priority_queue <pair<int,int> > q;
const int inf=2e9;
int n,m;
void read()
{
    in>>n>>m;
    dist.resize(n+1,inf);
    for(int i=0; i<m; i++)
    {
        int a,b,c;
        in>>a>>b>>c;
        v[a].push_back(make_pair(b,c));
    }
}
void dijkstra(int src)
{
    dist[src]=0;
    q.push({0,src});
    while(!q.empty())
    {
        int node=q.top().second;
        q.pop();
        //out<<node<<"\n";
        if(!viz[node])
        {
            viz[node]=1;
            for(int i=0; i<v[node].size(); i++)
            {
                if(!viz[v[node][i].first]  &&  dist[v[node][i].first] > dist[node]+v[node][i].second)
                {
                    dist[v[node][i].first]=dist[node]+v[node][i].second;
                    //out<<"i: "<<i<<" "<<v[node][i].first<<" "<<dist[v[node][i].first]<<"\n";
                    q.push({-dist[v[node][i].first],v[node][i].first});
                }
            }
        }
        //out<<q.top().second<<"\n";
    }
    //out<<"\n";
}
int main()
{
    read();
    dijkstra(1);
    for(int i=2; i<=n; i++)
        if(dist[i]==inf)
            out<<0<<" ";
        else
            out<<dist[i]<<" ";
    return 0;
}



//
//#include <bits/stdc++.h>
//
//using namespace std;
//
//const int NMAX = 5e4,INF = 1e9;;
//vector < pair <int, int> > e[NMAX + 5];
//priority_queue < pair<int, int> > pq;
//bool viz[NMAX + 5];
//int dist[NMAX + 5];
//
//void dijkstra(int src)
//{
//    int node,f,s;
//    dist[src] = 0;
//    pq.push({0, src});
//    while (!pq.empty())
//    {
//        node = pq.top().second;
//        pq.pop();
//        if (viz[node])
//            continue;
//        viz[node] = 1;
//        for (int i = 0; i < e[node].size(); i++)
//        {
//            f = e[node][i].first;
//            s = e[node][i].second;
//            if (!viz[f] && s + dist[node] < dist[f])
//            {
//                dist[f] = s + dist[node];
//                pq.push({-dist[f], f});
//            }
//        }
//    }
//}
//int main()
//{
//    ifstream fin("dijkstra.in");
//    ofstream fout("dijkstra.out");
//    int n,m,a,b,c,src = 1;
//    fin >> n >> m;
//    for (int i = 0; i < m; i++)
//    {
//        fin >> a >> b >> c;
//        e[a].push_back({b, c});
//    }
//    for (int i = 1; i <= n; i++)
//        dist[i] = INF;
//    dijkstra(src);
//    for (int i = 2; i <= n; i++)
//        if (dist[i] == INF)
//            fout << 0 << " ";
//        else
//            fout << dist[i] << " ";
//    return 0;
//}