Cod sursa(job #2869334)

Utilizator valentina_veleatVeleat Valentina-Georgiana valentina_veleat Data 11 martie 2022 14:11:25
Problema Arbore partial de cost minim Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.01 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("apm.in");ofstream g("apm.out");
short int a[20005][20005];int n,m,x,y,c,s[200005],v[200005],mn,p,nr,vf[200005],S;
void prim(int x)
{
    v[x]=1;
    vf[++nr]=x;
    s[x]=0;
    for(int k=1;k<n;k++)
    {
        mn=1001;
        for(int i=1;i<=nr;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(!v[j] && a[vf[i]][j]<mn)
                {
                    mn=a[vf[i]][j];
                    p=j;y=vf[i];
                }
            }
        }
        s[p]=y;
        v[p]=1;
        vf[++nr]=p;
        S+=mn;
    }
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(i!=j)a[i][j]=1005;
        }
    }
    for(int i=1;i<=m;i++)
    {
        f>>x>>y>>c;
        a[x][y]=a[y][x]=c;
    }
    prim(1);
    g<<S<<'\n'<<n-1<<'\n';
    for(int i=1;i<=n;i++)
    {
        if(s[i]!=0)g<<i<<" "<<s[i]<<'\n';
    }
    return 0;
}