Cod sursa(job #671018)

Utilizator robertpoeRobert Poenaru robertpoe Data 30 ianuarie 2012 16:29:11
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.84 kb
#include<fstream>
#include<algorithm>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
int n,m,i,j,costm,nr;
struct arbore
{
	int x,y,c;
	
};
arbore a[500001];
int compara(arbore a,arbore b)
{
	return a.c<b.c;
}
int d[500001];
int b[500001];
int det(int x)
{
	int r,i,y;
	r=x;
	while(d[r]!=r)
		r=d[r];
	while(x!=d[x])
	{
		y=d[x];
		d[x]=r;;
		x=y;
	}
	return r;
}
void calculeaza()
{
	int i;
	int m1,m2;
	for(i=1;i<=m&&nr<n-1;i++)
	{
		m1=det(a[i].x);
		m2=det(a[i].y);
		if(m1!=m2)
		{
			costm+=a[i].c;
			b[++nr]=i;
			d[m1]=m2;
		}
	}
}
int main()
{
	f>>n>>m;
	for(i=1;i<=m;i++)
		f>>a[i].x>>a[i].y>>a[i].c;
	for(i=1;i<=n;i++)
		d[i]=i;
	sort(a+1,a+m+1,compara);
	calculeaza();
	g<<costm<<"\n"<<nr<<"\n";
	for(i=1;i<=nr;i++)
		g<<a[b[i]].y<<" "<<a[b[i]].x<<"\n";
	return 0;
}