Pagini recente » Cod sursa (job #2427859) | Cod sursa (job #775491) | Cod sursa (job #520373) | Cod sursa (job #1468823) | Cod sursa (job #704206)
Cod sursa(job #704206)
#include <fstream>
#include <vector>
#include <algorithm>
#include <utility>
#define INF 32500
#define MAXN 50001
vector<pair<int,int> > graf[MAXN];
vector<pair<int,int> > heap;
int n,m;
int dist[MAXN];
bool viz[MAXN];
bool comp(pair<int,int>a,pair<int,int>b){ return a.second<b.second; }
int main(){
ifstream f("dijkstra.in");
f >> n >> m;
for(i=0;i<m;i++){
f>>a>>b>>c;
graf[a].push_back(make_pair(b,c));
graf[b].push_back(make_pair(a,c));
if(a==1) dist[b] = c;
else if(b==1) dist[a] = c;
else dist[a] = dist[b] = INF;
}
for(int i=2; i<=n;i++) heap.push_back(make_pair(i,dist[i]));
make_heap(v.begin(), v.end(), comp);
while(heap.size()){
int cur = heap[0].first;
vector<pair<int,int> >::iterator it;
for(it=graf[cur].begin(); it!=graf[cur].end(); it++){
if(!viz[it->first] && dist[it->first]>dist[cur]+it->second) dist[it->first]=dist[cur]+it->second;
}
viz[cur] = true;
heap.erase(heap.begin());
makeheap(v.begin(), v.end(), comp);
}
ofstream g("dijkstra.out");
for(int i=2;i<=n;i++) g << dist[i]!=INF?dist[i]:0 << " ";
g.close();
f.close();
return 0;
}