Cod sursa(job #2677545)

Utilizator cyg_CiuntuSorinCiuntu Sorin Andrei cyg_CiuntuSorin Data 26 noiembrie 2020 19:49:07
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.4 kb
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;

struct MUCHIE
{
    int u,v,c,sel;
};

vector<MUCHIE>v;
vector<int>t,h;

int FindSet(int x)
{
    while(x!=t[x])
        x=t[x];
    return x;
}

void UnionSet(int x,int y)
{
    if(h[x]==h[y])
    {
        t[y]=x;
        h[x]++;
    }
    else if(h[x]>h[y])
        t[y]=x;
    else
        t[x]=y;
}

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

int main()
{
    int n,m,i,k,tc,tu,tv;
    MUCHIE temp;
    scanf("%d%d",&n,&m);
    t.assign(n+1,0);
    h.assign(n+1,1);
    for(i=1;i<=n;i++)
        t[i]=i;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d",&temp.u,&temp.v,&temp.c);
        temp.sel=0;
        v.push_back(temp);
    }
    sort(v.begin(),v.begin()+m,cmp);
    tc=k=0;
    for(i=0;i<m && k<n-1;i++)
    {
        tu=FindSet(v[i].u);
        tv=FindSet(v[i].v);
        if(tu!=tv)
        {
            UnionSet(tu,tv);
            k++;
            tc=tc+v[i].c;
            v[i].sel=1;
        }
    }
    printf("%d\n",tc);
    int nrsel=0;
    for(i=0;i<v.size();i++)
        if(v[i].sel)
            nrsel++;
    printf("%d\n",nrsel);
    for(i=0;i<v.size();i++)
        if(v[i].sel)
            printf("%d %d\n",v[i].u,v[i].v);
    return 0;
}


/*

9 11
1 2 2
1 7 4
2 3 3
2 5 2
2 6 3
2 7 3
3 4 1
3 5 2
4 5 1
5 6 3
6 7 5

*/