Cod sursa(job #1434049)

Utilizator alex.vasiuVasiu Alexandru alex.vasiu Data 10 mai 2015 12:24:53
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <iterator>
#include <algorithm>
#include <climits>
#include <ctime>
#include <sstream>
using namespace std;

struct xxx {int a,b,c;};
xxx sol[400005],l[400005];
int s(xxx x,xxx y)
{
    if(x.c > y.c)
        return 0;
    return 1;
}
char solutie[600010];
ifstream in ("apm.in");
int n, m;
int i, j, ct,k=0;
int v[200005], f[400005];
int main()
{
    in >> n >> m;
    int a,b,c;
    xxx q;
    q.a=q.b=q.c=-INT_MAX+1;
    l[0]=q;
    for(i=1; i<=m; i++)
    {
        in>>l[i].a>>l[i].b>>l[i].c;
        //l.push_back(q);
    }
    make_heap(l+1,l+m+1,s);
    sort_heap(l+1,l+m+1,s);
    v[1]=1;
    ofstream g("am.out");
    for(i=1; i<=m; i++)
    {
        a=l[i].a;
        b=l[i].b;
        c=l[i].c;
        if(!f[i])
        {
            if(v[a] + v[b] ==1)
            {
                ct+=c;
                f[i]++;
                v[a]=v[b]=1;
                i=0;
               g<<a<<" "<<b<<"\n";
            }
        }
    }
    g.close();
    ifstream fin("am.out");
    ofstream gi("apm.out");
    gi  << ct<<"\n"<<n-1<<"\n";
    while(fin.getline(solutie,600010))
    gi<<solutie<<"\n";
}