Cod sursa(job #374358)

Utilizator cosmin79Carabet Cosmin Andrei cosmin79 Data 16 decembrie 2009 20:32:45
Problema Pitici Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.01 kb
#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;
}