Cod sursa(job #696207)

Utilizator ms-ninjacristescu liviu ms-ninja Data 28 februarie 2012 17:35:17
Problema Arbore partial de cost minim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.89 kb
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
#define dim 100010
int viz[dim*2],n , m, nrm, cost, t[dim*2], r[dim*2];
ifstream fin("apm.in");
ofstream fout("apm.out");

struct muchie{int x, y, c;}v[dim*4], sol[dim];

inline int cmp(muchie a, muchie b)
{
	return a.c<b.c;
}

void read()
{
	fin>>n >>m;
	for(int i=1;i<=m;++i)
		fin>>v[i].x >>v[i].y >>v[i].c;
}

int find(int x)
{
	if(x!=t[x])
		t[x]=find(t[x]);
	return t[x];
}

inline void unite(int x, int y)
{
	if(r[x]<r[y])
		t[x]=y;
	else
		t[y]=x;
	if(r[x]==r[y])
		++r[x];
}

void solve()
{
	int i;
	
	for(i=1;nrm<n-1 && i<=m;++i)
		if( find(v[i].x)!=find(v[i].y) )
		{
			cost+=v[i].c;
			sol[++nrm]=v[i];
			unite(v[i].x,v[i].y);
		}
}
	
int main()
{
	read();
	for(int i=1;i<=n;++i)
		t[i]=i;
	sort(v+1,v+n+1,cmp);
	solve();
	
	return 0;
}