Cod sursa(job #1992599)

Utilizator patcasrarespatcas rares danut patcasrares Data 20 iunie 2017 22:00:45
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.53 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;
int M;
int dist[DN],expe[DN];
vector<pair<int,int > >gr[DN];
typedef pair<int,int >pii;
struct cmp
{
    bool operator()(pii A,pii B)
    {
        return A.first<B.first;
    }
};
set<pair<int,int>,cmp >s;
void ve()
{
    int i;
    int mi=M,ne;
    for(i=2;i<=n;i++)
    {
        dist[i]=M;
        s.insert(make_pair(M,i));
    }
    dist[1]=0;
    s.insert(make_pair(0,1));
    while(!s.empty())
    {
        ne=s.begin()->second;
        mi=s.begin()->first;
        expe[ne]=1;
        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(make_pair(dist[vecin],vecin));
                    dist[vecin]=dist[ne]+cost;
                    if(expe[vecin]==0)
                        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));
    }
    M=20000*250000;
    ve();
    for(int i=2;i<=n;i++)
        if(dist[i]!=M)
            fout<<dist[i]<<' ';
        else fout<<0<<' ';
}