Cod sursa(job #3210968)

Utilizator biancalautaruBianca Lautaru biancalautaru Data 7 martie 2024 20:04:19
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>
#include <algorithm>
#define NMAX 200001
#define MMAX 400001
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int n,m,rx,ry,cnt,sum,t[NMAX];
struct muchie {
    int x,y,c;
    bool viz;
}v[MMAX];

bool cmp(muchie a,muchie b) {
    return a.c<b.c;
}

int root(int x) {
    while (t[x]>0)
        x=t[x];
    return x;
}

int main() {
    fin>>n>>m;
    for (int i=1;i<=m;i++)
        fin>>v[i].x>>v[i].y>>v[i].c;
    sort(v+1,v+m+1,cmp);
    for (int i=1;i<=n;i++)
        t[i]=-1;
    int i=1;
    while (cnt<n-1) {
        int rx=root(v[i].x);
        int ry=root(v[i].y);
        if (rx!=ry) {
            if (-t[rx]>-t[ry]) {
                t[rx]+=t[ry];
                t[ry]=rx;
            }
            else {
                t[ry]+=t[rx];
                t[rx]=ry;
            }
            cnt++;
            sum+=v[i].c;
            v[i].viz=1;
        }
        i++;
    }
    fout<<sum<<"\n"<<cnt<<"\n";
    for (int i=1;i<=m;i++)
        if (v[i].viz==1)
            fout<<v[i].x<<" "<<v[i].y<<"\n";
    return 0;
}