Pagini recente » Cod sursa (job #930922) | Cod sursa (job #2931284) | Cod sursa (job #1955447) | Cod sursa (job #2212182) | Cod sursa (job #2153434)
#include <fstream>
#include <vector>
#include <set>
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
int N,M,dist[50005];
const int INF = 0x3f3f3f3f;
vector <pair< int, int> > v[50005];
set <pair<int, int> > s;
vector<pair<int, int> > ::iterator it;
void citire()
{
//citim n,m si apoi m linii
f>>N>>M;
int x,y,cost;
for(int i=1;i<=M;i++)
{
//adaugam intr-un vector de perechi care va tine minte vecinul si costul
f>>x>>y>>cost;
v[x].push_back(make_pair(y,cost));
}
}
void Djikstra(int nodInitial)
{
int nodCurent,vecin,cost;
//toate dinstantele devin infinit
for(int i=1;i<=N;i++)
dist[i]=INF;
//distanta de la nodul initial la nodul initial este 0
dist[nodInitial]=0;
//adaugam nodul initial in setul cate va tine minte cel mai scurt drum
s.insert(make_pair(nodInitial,0));
//cat timp mai sunt elemente in set
while(!s.empty())
{
//nod curent ia valoarea primului nod din set(ordine crescatoare)
nodCurent=s.begin()->first;
//stergem primul nod din set
s.erase(s.begin());
//pentru fiecare vecin al nodului curent
for(it=v[nodCurent].begin();it!=v[nodCurent].end();it++)
{
vecin=it->first;
cost=it->second;
//daca noua distanta este mai buna decat cea veche
if(dist[vecin]>dist[nodCurent]+cost)
{
//daca s-a mai ajuns in nodul acesta se sterge din set
if(dist[vecin]!=INF)
s.erase( s.find( make_pair(vecin,dist[vecin]) ) );
//distanta se actualizeaza iar apoi se adauga in set nodul;
dist[vecin]=dist[nodCurent]+cost;
s.insert( make_pair(vecin,dist[vecin]) );
}
}
}
}
void afisare()
{
//afisarea distantei de la nodul cerut la toate celalalte noduri
for(int i=2;i<=N;i++)
{
if(dist[i]==INF)g<<0<<' ';
else g<<dist[i]<<' ';
}
}
int main()
{
citire();
Djikstra(1);
afisare();
f.close();
g.close();
return 0;
}