Cod sursa(job #1496125)

Utilizator adina0822Ciubotaru Adina-Maria adina0822 Data 4 octombrie 2015 14:08:15
Problema Arbore partial de cost minim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.88 kb
using namespace std;
#include <fstream>
#define NMaxVf 200000
#define NMaxMuchii 400000
ifstream f ("apm.in");
ofstream g ("apm.out");

struct Muchie {int e1,e2,cost;};
Muchie G[NMaxMuchii],y[NMaxMuchii];
int A[NMaxVf], c[NMaxVf],costapm=0;
int n,m;

void inter(int st,int mij,int dr)
{
    int i=st,j=mij+1,k=1;
    while(i<=mij && j<=dr)
    {
        if(G[i].cost<G[j].cost)
        {
            y[k]=G[i];
            i++;
        }
        else
        {
            y[k]=G[j];
            j++;
        }
        k++;
    }
    if(i<=mij) for(j=i; j<=mij; j++) {y[k]=G[j]; k++;}
    else for(i=j; i<=dr; i++){y[k]=G[i]; k++;}

    k=1;
    for(i=st;i<=dr;i++)
    {
        G[i]=y[k];
        k++;
    }

}


void merge(int st, int dr)
{
    Muchie aux;
    if(dr-st<=1)
    {
        if(G[st].cost>G[dr].cost)
        {
            aux=G[st];
            G[st]=G[dr];
            G[dr]=aux;
        }
    }
    else
    {
        int mij=(st+dr)/2;
        merge(st,mij);
        merge(mij+1,dr);
        inter(st,mij,dr);
    }
}

void afisare()
{
    int i;
    g<<costapm<<'\n';
    g<<n-1<<'\n';
    for(i=1; i<n; i++)
    g<<G[A[i]].e1<<" "<<G[A[i]].e2<<'\n';
}

int main()
{
  int i,j,min,max,NrMSel=0;
  f>>n>>m;
  for(i=1; i<=m; i++)
    f>>G[i].e1>>G[i].e2>>G[i].cost;
  for(i=1; i<=n; i++) c[i]=i;

  merge(1,m);
  //for(i=1; i<=m; i++) g<<G[i].e1<<" "<<G[i].e2<<" "<<G[i].cost<<'\n';

  for(i=1; NrMSel<n-1; i++)
    if(c[G[i].e1]!=c[G[i].e2])
    {
        A[++NrMSel]=i;
        costapm+=G[i].cost;
        if(c[G[i].e1]<c[G[i].e2])
        {
            min=c[G[i].e1];
            max=c[G[i].e2];
        }
        else
        {
            min=c[G[i].e2];
            max=c[G[i].e1];
        }
        for(j=1; j<=n; j++)
          if(c[j]==max) c[j]=min;
    }
  afisare();

  return 0;
}