Cod sursa(job #2294055)

Utilizator cezar.plescaCezar Plesca cezar.plesca Data 1 decembrie 2018 20:58:25
Problema Arbore partial de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.57 kb
#include<stdio.h>
#include<stdlib.h>

#include<set>
#include<list>
#include<queue>

#include <fstream>
#include <iostream>

using namespace std;

#define MAXN 200000
#define MAXM 400000

#define INFINIT 1001
int M,N;
bool ciclu;

list<pair<int,int>> L[MAXN];
set<pair<int,int>> Q; 
queue<pair<int,int>> F;

bool outQ[MAXN];
int C[MAXN];
int E[MAXN];

ifstream fin("apm.in");
ofstream fout("apm.out");

int costtotal;

void arboreprim(){
	for(int i=0;i<N;i++){
		Q.insert(make_pair(INFINIT,i));
		C[i]=INFINIT;
		E[i]=-1;
	}

	int nod,vecin,cost;
	set<pair<int,int>>::iterator itS,it; 
	list<pair<int,int>>::iterator itL;

	while(!Q.empty()){
		itS=Q.begin();
		nod=itS->second;
		if(E[nod]!=-1){			
			// adaugare muchie nod,vecin
			F.push(make_pair(nod,E[nod]));
			costtotal+=C[nod];
		}
		Q.erase(itS);

		for(itL=L[nod].begin();itL!=L[nod].end();itL++){
			vecin=itL->first;
			cost=itL->second;	
			if(!outQ[vecin] && cost<C[vecin]){
				it=Q.find(make_pair(C[vecin],vecin));
				C[vecin]=cost;
				E[vecin]=nod;
				Q.erase(it);
				Q.insert(make_pair(C[vecin],vecin));
			}
		}
		outQ[nod]=1;	
	}
}

int main(){
		
	fin >> N >> M;

	int x,y,c;

	for(int i=0;i<M;i++){
		fin >> x >> y >> c;
		L[x-1].push_back(make_pair(y-1,c));
		L[y-1].push_back(make_pair(x-1,c));
	}

	arboreprim();

	fout << costtotal << endl;
	fout << F.size() << endl;

	while(!F.empty()){
		x=F.front().first;
		y=F.front().second;
		fout << x+1 << " " << y+1 << endl;
		F.pop();
	}

	fin.close();
    fout.close();

	return 0;
}