Cod sursa(job #664746)

Utilizator okros_alexandruOkros Alexandru okros_alexandru Data 20 ianuarie 2012 18:53:36
Problema Arbore partial de cost minim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 0.78 kb
#include<fstream>
#include<algorithm>
#define NMAx 200100
using namespace std;
int x[NMAx],y[NMAx],C[NMAx],v[NMAx],T[NMAx],APM[NMAx];
int n,m,sol,nr;

bool cmp(int a,int b) {
	return C[a]<C[b];
}
int tata(int nod) {
	if(T[nod]!=nod)
		return tata(T[nod]);
	return nod;
}
void citire() {
	int i;
	ifstream in("apm.in");
	in>>n>>m;
	for(i=1;i<=m;i++) {
		in>>x[i]>>y[i]>>C[i];
		T[i]=v[i]=i;
		}
	in.close();
}
void afis() {
	int i;
	ofstream out("apm.out");
	out<<sol<<'\n'<<n-1<<'\n';
	for(i=0;i<n-1;i++)
		out<<x[APM[i]]<<" "<<y[APM[i]]<<'\n';
}
int main() {
	int i,a,b;
	citire();
	sort(v+1,v+m+1,cmp);
	for(i=1;i<=m;i++) {
		a=tata(x[v[i]]);
		b=tata(y[v[i]]);
		if(a!=b) {
			sol+=C[v[i]];
			T[a]=b;
			APM[nr++]=v[i];
			}
		}
	afis();
	return 0;
}