Cod sursa(job #2018904)

Utilizator vic2002Melinceanu Victor vic2002 Data 6 septembrie 2017 12:30:37
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#pragma GCC optimize("O3")
#include<bits/stdc++.h>
#define rc(x) return cout<<x<<endl,0
#define pb push_back
#define in insert
#define er erase
#define fr first
#define sc second
typedef long long ll;
const long long llinf=LLONG_MAX;
const int inf=INT_MAX;
const int nmax=5e4+5;
const int mod=1e9+7;
using namespace std;
int n,m,i,x,y,nxt;
ll dist[nmax],c,cs;
vector<pair<int,ll> >a[nmax];
set<pair<ll,int> >s;
int main()
{
	freopen("dijkstra.in","r",stdin);
	freopen("dijkstra.out","w",stdout);
	ios_base::sync_with_stdio(false);cin.tie(0);cerr.tie(0);cout.tie(0);
	cin>>n>>m;
	for(i=2;i<=n;i++)dist[i]=inf;
	while(m--)
	{
		cin>>x>>y>>c;
		a[x].pb(make_pair(y,c));
	}
	s.in(make_pair(0,1));
	while(!s.empty())
	{
		cs=s.begin()->fr;
		nxt=s.begin()->sc;
		s.er(s.begin());
		if(cs!=inf)
		 for(i=0;i<a[nxt].size();i++)
		  if(dist[nxt]+a[nxt][i].sc<dist[a[nxt][i].fr])
		  {
		  	if(dist[a[nxt][i].fr]!=inf)s.er(s.find(make_pair(dist[a[nxt][i].fr],a[nxt][i].fr)));
		  	dist[a[nxt][i].fr]=dist[nxt]+a[nxt][i].sc;
		  	s.in(make_pair(dist[a[nxt][i].fr],a[nxt][i].fr));
		  }
	}
	for(i=2;i<=n;i++)
     if(dist[i]!=inf)cout<<dist[i]<<" ";
     else cout<<"0 ";
    cout<<endl;
	return 0;
}