Cod sursa(job #1369886)

Utilizator bianncaPoenar Bianca biannca Data 3 martie 2015 11:58:57
Problema Arbore partial de cost minim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.52 kb
//#include <iostream>
#include<fstream>
#include<vector>
#include<cassert>
#include<algorithm>
//#include<stdio.h>
#define mx 400001
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
vector<int> v[mx], vect1[mx],vect2[mx];
vector<pair<int, int> >  sf;
vector<pair<int,pair<int,int> > > vect;
int viz[mx];
int n,m,c,tot=0,x,y;
void citire()
{
    int x,y,i;
    fin>>n>>m;
        for(i=1;i<=m;i++)
        {
            fin>>x>>y>>c;
            v[x].push_back(y);
            v[y].push_back(x);
            vect.push_back(make_pair(c,make_pair(x,y)));
        }
fin.close();
}
void dfs(int nod)
{
    viz[nod]=1;
    for(unsigned int i=0;i<vect1[nod].size();++i)
        if(viz[vect1[nod][i]]==0)
        dfs(vect1[nod][i]);
}
void sl()
{
    int i,j;

    sort(vect.begin(),vect.end());
    int nr=n;
    for(i=0;i<m;++i)
    {
        int cost=vect[i].first;
        int nodx=vect[i].second.first;
        int nody=vect[i].second.second;
       //memset(viz,0,sizeof(viz));
       for(int k= 0 ; k < m ; ++ k)
        viz[i]=0;
        dfs(nodx);
        if(viz[nody]==false)
        {
            nr--;
            tot+=cost;
            vect1[x].push_back(y);
            vect2[y].push_back(x);
            sf.push_back(make_pair(nody,nodx));
        }
        if(nr==1) {break;}
    }
    fout<<tot<<'\n';
    fout<<n-1<<'\n';
    for(i=0;i<n-1;i++)
        fout<<sf[i].first<<" "<<sf[i].second<<'\n';
}

int main()
{
    citire();
    sl();
        return 0;
}