Cod sursa(job #1992613)

Utilizator patcasrarespatcas rares danut patcasrares Data 20 iunie 2017 22:25:47
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.32 kb
#include<fstream>
#include<queue>
#include<algorithm>
#include<vector>
#include<set>
#define DN 50005
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
int n,m,a,b,c;
const int M = 0x3f3f3f3f;
int dist[DN],expe[DN];
vector<pair<int,int > >gr[DN];
typedef pair<int,int >pii;
set<pair<int,int> >s;
void ve()
{
    int i;
    for(int i=2;i<=n;i++)
        dist[i]=M;
    dist[1]=0;
    s.insert(make_pair(0,1));
    while(!s.empty())
    {
        int ne=s.begin()->second;
        int mi=s.begin()->first;
        s.erase(s.begin());
        if(mi!=M)
            for(i=0;i<gr[ne].size();i++)
            {
                int vecin=gr[ne][i].first;
                int cost=gr[ne][i].second;
                if(dist[ne]+cost<dist[vecin])
                {
                    if(dist[vecin]!=M)
                        s.erase(s.find(make_pair(dist[vecin],vecin)));
                    dist[vecin]=dist[ne]+cost;
                    s.insert(make_pair(dist[vecin],vecin));
                }
            }

    }
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>a>>b>>c;
        gr[a].push_back(make_pair(b,c));
    }
    ve();
    for(int i=2;i<=n;i++)
        if(dist[i]!=M)
            fout<<dist[i]<<' ';
        else fout<<0<<' ';
}