Cod sursa(job #2680136)

Utilizator razvanmihailaMihaila Razvan razvanmihaila Data 2 decembrie 2020 18:53:45
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.24 kb
#include<iostream>
#include<fstream>
#include<algorithm>
#define ll long long
#define identifier 1001
using namespace std;

ll tata[200001], nrmuchii, s;

struct mama
{
    int x, y, cost;
}v[400001];

int findtata(int x)
{
    if(x == tata[x])
        return x;
    else
        return tata[x] = findtata(tata[x]);
}

void uniune(int x, int y)
{
    int sefx, sefy;
    sefx = findtata(x);
    sefy = findtata(y);
    tata[sefx] = sefy; 
}

bool cmp(mama a, mama b)
{
    return a.cost < b.cost;
}
int main()
{

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

    int i, n, m;

    fin >> n >> m;

    for(i = 1; i <= n; i ++)
        tata[i] = i;

    for(i = 1; i <= m; i ++)
        fin >> v[i].x >> v[i].y >> v[i].cost;

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

    for(i = 1; i <= m; i ++)
    {
        if(findtata(v[i].x) != findtata(v[i].y))
        {
            uniune(v[i].x, v[i].y);
            nrmuchii ++;
            s += v[i].cost;
            v[i].cost = identifier;
        }
    }
    fout << s << '\n';
    fout << nrmuchii << '\n';
    for(i = 1; i <= m; i ++)
        if(v[i].cost == identifier)
            fout << v[i].x << " " << v[i].y << '\n';

    return 0;
    
}