Pagini recente » Cod sursa (job #1463164) | Cod sursa (job #2805269) | Cod sursa (job #2868067) | Cod sursa (job #289132) | Cod sursa (job #575233)
Cod sursa(job #575233)
#include <utility>
#include <functional>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;
#define N 50001
vector<pair<int,int> > g[N];//vecinul, costul
bool v[N];
int d[N];
struct Compare
{ bool operator()(const int& a, const int& b)
{ return d[a]>d[b];
}
};
int main ()
{ freopen("dijkstra.in","r",stdin);
freopen("dijkstra.out","w",stdout);
int i,j,n,m,a,b,c,mv,iv,ic;
scanf("%d %d",&n,&m);
for (i=1;i<=m;i++)
{ scanf("%d %d %d",&a, &b, &c);
g[a].push_back(make_pair(b,c));
}
vector<int> h;
d[1]=0;
v[1]=true;
for (i=0;i<g[1].size();i++)
{ iv=g[1][i].first;
ic=g[1][i].second;
h.push_back(iv);
v[iv]=true;
d[iv]=ic;
}
make_heap(h.begin(),h.end(),Compare());
for(i=1;i<n;i++)
{ mv=h[0];
pop_heap(h.begin(),h.end());
h.pop_back();
for (j=0;j<g[mv].size();j++)
{ iv=g[mv][j].first;
ic=g[mv][j].second;
if(v[iv])
{ if(d[iv]>d[mv]+ic)
{ d[iv]=d[mv]+ic;
}
}
else
{ d[iv]=d[mv]+ic;
v[iv]=true;
h.push_back(iv);
}
}
make_heap(h.begin(),h.end(),Compare());
}
for (i=2;i<=n;i++)
{ printf("%d ",d[i]);
}
return 0;
}