Cod sursă (job #675972)

Utilizator avatar carpstefan Carp Stefan carpstefan IP ascuns
Problemă Zapada (clasele 11 și 12) Compilator cpp-32 | 1,32 kb
Rundă vaslui_cls1112_15.11 Status evaluat
Dată 15 nov. 2022 19:39:28 Scor 0
#include <bits/stdc++.h>
#define nmax 100005

using namespace std;

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

/**
6 9
1 2 3
1 3 1
1 4 8
1 6 3
2 3 4
3 4 3
4 5 4
4 6 6
5 6 2
*/
struct Muchie
{
    int x,y,cost;
};

int t[nmax],n,m,sol,k;
Muchie a[100005];

bool Cmp(Muchie A,Muchie B)
{
    return A.cost<B.cost;
}

/// unifica radacinile x si y
void Union(int x,int y)
{
    t[y]=x;
}

/// Complexitate O(log*n)
int Find(int x)
{
    int y,rad=x;
    while(t[rad]!=0)
        rad=t[rad];
    /// compresie drum
    while(x!=rad)
    {
        y=t[x];
        t[x]=rad;
        x=y;
    }
    return rad;
}
int Rezolvare(Muchie a[],int m)
{
    int i,j,y,x;
    sol=0;
    sort(a+1,a+m+1,Cmp);
    for (i=1;i<=m;i++)
    {
        x=Find(a[i].x);
        y=Find(a[i].y);
        if (x!=y)
        {
            Union(x,y);
            sol+=a[i].cost;
        }
    }
    return sol;
}
int main()
{
    int i,x,y,z;
    fin>>n>>m>>k;
    for (i=1;i<=m;i++)
        fin>>a[i].x>>a[i].y>>a[i].cost;
    fout<<Rezolvare(a,m)<<"\n";
    for(i=1;i<=k;i++)
    {
        fin>>x>>y>>z;
        m++;
        a[m].x=x;
        a[m].y=y;
        a[m].cost=z;
        fout<<Rezolvare(a,m)<<"\n";
    }
    fin.close();
    fout.close();
    return 0;

}