Cod sursa(job #761943)

Utilizator vendettaSalajan Razvan vendetta Data 27 iunie 2012 21:56:41
Problema Lazy Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.82 kb
#include<fstream>
#define dim 200001
#include<algorithm>
using namespace std;


ifstream f("lazy.in");
ofstream g("lazy.out");
int T[dim],x[dim],y[dim],I[dim],n,m;
long long pr[dim],cost[dim];
inline int tata(int x){
	if(T[x]==x)
		return x;
	T[x]=tata(T[x]);

	return T[x];
}
struct cmp
{
	bool operator() (const int &a, const int &b)const
	{
		if (cost[a]==cost[b])
			return pr[a]>pr[b];
		else
			return cost[a]<cost[b];
	}
};
void uni(int i,int j){

	T[tata(i)]=tata(j);
}
int main () {

	f>>n>>m;
	int a,b ;
	for(int i=1;i<=m;i++){
		f>>x[i]>>y[i]>>cost[i]>>pr[i];
		I[i]=i;
	}
	for(int i=1;i<=n;i++)
		T[i]=i;

	sort(I+1,I+1+m,cmp());

	for(int i=1;i<=m;i++){

		a=tata(x[I[i]]);
		b=tata(y[I[i]]);

		if(a!=b){
			g<<I[i]<<"\n";
			uni(x[I[i]],y[I[i]]);

		}


	}
	return 0;
}