Pagini recente » Cod sursa (job #832809) | Cod sursa (job #485319) | Cod sursa (job #2196286) | Cod sursa (job #974383) | Cod sursa (job #374358)
Cod sursa(job #374358)
#include <stdio.h>
#include <string.h>
#include <vector>
#include <set>
#include <queue>
#define mp make_pair
#define pb push_back
#define f first
#define s second
#define N 1<<10
#define inf 1000000000
#define ii pair<int, int>
using namespace std;
int n,m,k,sz;
vector <ii> A[N];
int cost[N][N],pre[N],nrst[N];
char viz[N];
struct heap
{
int a,b;
};
heap v[N];
void read()
{
scanf("%d%d%d",&n,&m,&k);
int i,x,y,z;
for (i=1; i<=m; i++)
{
scanf("%d%d%d",&x,&y,&z);
A[x].pb(mp(y,z));
}
}
void init()
{
int i,j;
for (i=1; i<=n; i++)
for (j=1; j<=k; j++)
cost[i][j]=inf;
cost[n][1]=0;
}
void interschimba(int poz1, int poz2)
{
int temp = v[poz1].a,temp2=v[poz1].b;
v[poz1].a = v[poz2].a;
v[poz1].b = v[poz2].b;
v[poz2].a = temp;
v[poz2].b = temp2;
}
void mut_in_sus(int poz)
{
if (poz > 1 && v[poz/2].a > v[poz].a)
{
interschimba(poz, poz/2);
mut_in_sus(poz/2);
}
}
void mut_in_jos(int poz)
{
if ((2*poz <= sz && v[poz].a > v[2*poz].a ) ||
(2*poz+1<= sz && v[poz].a > v[2*poz+1].a))
{
int minimum = 2*poz;
if (2*poz+1 <= sz && v[2*poz+1].a < v[2*poz].a)
minimum= 2*poz+1;
interschimba(poz, minimum);
mut_in_jos(minimum);
}
}
void del()
{
v[1].a=v[sz].a;
v[1].b=v[sz].b;
sz--;
mut_in_jos(1);
}
void dfs(int nod)
{
if (viz[nod])
return ;
viz[nod]=1;
int i,st,val,x;
for (i=0; i<A[nod].size(); i++)
dfs(A[nod][i].f);
sz=0;
for (i=0; i<A[nod].size(); i++)
{
st=A[nod][i].f;
pre[st]=1;
nrst[st]=A[nod][i].s;
v[++sz].a=cost[st][1]+A[nod][i].s;
v[sz].b=st;
mut_in_sus(sz);
}
int r=0;
for (i=1; i<=k && sz; i++)
{
val=v[1].a,x =v[1].b;
del();
if (val>=inf)
break;
cost[nod][++r]=val;
++pre[x];
if (pre[x]<=k)
{
v[++sz].a=nrst[x]+cost[x][pre[x]];
v[sz].b=x;
mut_in_sus(sz);
}
}
}
void show()
{
int i;
for (i=1; i<=k; i++)
printf("%d ",cost[1][i]);
printf("\n");
}
int main()
{
freopen("pitici.in","r",stdin);
freopen("pitici.out","w",stdout);
read();
init();
dfs(1);
show();
return 0;
}