Pagini recente » Cod sursa (job #2454493) | Cod sursa (job #744922) | Cod sursa (job #1041031) | Cod sursa (job #2599537) | Cod sursa (job #1397769)
# include <fstream>
# include <algorithm>
# include <vector>
# include <queue>
# define mp make_pair
# define NR 1050
using namespace std;
ifstream f("pitici.in");
ofstream g("pitici.out");
vector <int> v[NR];
vector <pair <int, pair<int, int> > > HEAP; //valoare, I, L[i]
int i,j,n,m,x,y,w,K,cost,k,nr,ant,l,urm;
int ord[NR], ap[NR], L[NR], a[NR][NR];
short C[NR][NR];
void DFS (int k)
{
ap[k]=1;
for (int i=0; i<v[k].size(); ++i)
if (! ap[v[k][i]]) DFS (v[k][i]);
ord[++w]=k;
}
int main ()
{
f>>n>>m>>K;
for (i=1; i<=m; ++i)
{
f>>x>>y>>cost; C[x][y]=C[y][x]=cost;
v[y].push_back(x);
}
DFS (n); L[1]=1; a[1][1]=0;
for (i=2; i<=n; ++i)
{
k=ord[i]; HEAP.clear();
for (j=0; j<v[k].size(); ++j)
{
int urm=v[k][j];
if (L[urm])
{
HEAP.push_back(mp(-(a[urm][1] + C[k][urm]), mp (urm, 1)));
push_heap(HEAP.begin(), HEAP.end());
}
}
while (HEAP.size() && L[k]<=K)
{
nr=-HEAP[0].first;
ant=HEAP[0].second.first;
l=HEAP[0].second.second;
pop_heap(HEAP.begin(), HEAP.end());
HEAP.pop_back();
a[k][++L[k]]=nr;
if (l<L[ant])
{
HEAP.push_back(mp(-(a[ant][++l] + C[k][ant]), mp (ant, l)));
push_heap(HEAP.begin(), HEAP.end());
}
}
}
for (i=1; i<=K; ++i)
g<<a[n][i]<<" ";
g<<"\n";
return 0;
}