Cod sursa(job #545754)

Utilizator maritimCristian Lambru maritim Data 3 martie 2011 21:16:13
Problema Lazy Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.44 kb
#include<stdio.h>
#include<malloc.h>
using namespace std;

typedef struct 
{
	long int x;
	long int y;
	long long cost;
	long long val;
	long int poz;
} xy;

xy M[200001];
long int n;
long int m;
long int L[200001];

void citire(void)
{
	FILE *f = fopen("lazy.in","r");
	
	fscanf(f,"%d %d",&n,&m);
	for(int i=1;i<=m;M[i].poz = i,i++)
		fscanf(f,"%d %d %d %d",&M[i].x,&M[i].y,&M[i].cost,&M[i].val);
	
	fclose(f);
}

long int poz(int li,int ls)
{
	int i1 = 0;
	int j1 = -1;
	int jiaux = 0;
	xy c = M[(int)(li+ls)/2];
	M[(int)(li+ls)/2] = M[li];
	M[li] = c;
	while(li<ls)
	{
		if(M[li].cost>M[ls].cost || (M[li].cost == M[ls].cost && M[li].cost*M[li].val <M[ls].cost*M[ls].val))
		{
			c = M[li];
			M[li] = M[ls];
			M[ls] = c;
			jiaux = -i1;
			i1 = - j1;
			j1 = jiaux;
		}
		li += i1;
		ls += j1;
	}
	return li;
}

void quick(int li,int ls)
{
	if(li<ls)
	{
	long int k = poz(li,ls);
	quick(li,k-1);
	quick(k+1,ls);
	}
}

void complet(void)
{
	for(int i=1;i<=m;i++)
		L[i] = i;
}

void kruscal(void)
{
	FILE *f = fopen("lazy.out","w");
	
	long int nr = 0;
	long int k = 1;
	long int pz = 0;
	while(nr < n-1)
	{
		if(L[M[k].x] != L[M[k].y])
		{
			fprintf(f,"%d \n",M[k].poz);
			pz = L[M[k].y];
			for(int i=1;i<=m;i++)
				if(L[i] == pz)
					L[i] = L[M[k].x];
			nr ++;
		}
		k ++;
	}
	
	fclose(f);
}

int main()
{
	citire();
	quick(1,m);
	complet();
	kruscal();
	return 0;
}