Pagini recente » Cod sursa (job #1760978) | Cod sursa (job #1177087) | Cod sursa (job #2238097) | Cod sursa (job #2765108) | Cod sursa (job #545813)
Cod sursa(job #545813)
#include<stdio.h>
#include<malloc.h>
using namespace std;
typedef struct
{
long int x;
long int y;
long int cost;
long int val;
long int poz;
} xy;
xy M[200001];
long int n;
long int m;
long int L[200001];
char T[200001];
long long S;
void citire(void)
{
FILE *f = fopen("lazy.in","r");
fscanf(f,"%d %d",&n,&m);
for(int i=1;i<=m;i++)
{
fscanf(f,"%d %d %d %d",&M[i].x,&M[i].y,&M[i].cost,&M[i].val);
M[i].poz = i;
}
fclose(f);
}
long int poz(int li,int ls)
{
int i1 = 0;
int j1 = -1;
int jiaux = 0;
xy c = M[(int)(li+ls)/2];
M[(int)(li+ls)/2] = M[li];
M[li] = c;
while(li<ls)
{
if(M[li].cost>M[ls].cost || (M[li].cost == M[ls].cost && M[li].val < M[ls].val))
{
c = M[li];
M[li] = M[ls];
M[ls] = c;
jiaux = -i1;
i1 = - j1;
j1 = jiaux;
}
li += i1;
ls += j1;
}
return li;
}
void quick(int li,int ls)
{
if(li<ls)
{
long int k = poz(li,ls);
quick(li,k-1);
quick(k+1,ls);
}
}
void complet(void)
{
for(int i=1;i<=m;i++)
L[i] = i;
}
void kruscal(void)
{
FILE *f = fopen("lazy.out","w");
long int nr = 0;
long int k = 1;
long int pz = 0;
long int po = 0;
while(nr < n-1)
{
if(L[M[k].x] != L[M[k].y])
{
S += M[k].cost*M[k].val;
fprintf(f,"%d \n",M[k].poz);
pz = L[M[k].y];
po = L[M[k].x];
for(int i=1;i<=m;i++)
if(L[i] == pz)
L[i] = po;
nr ++;
}
k ++;
}
fclose(f);
}
int main()
{
citire();
quick(1,m);
complet();
kruscal();
// for(int i=1;i<=m;i++)
// printf("%d %d\n",M[i].cost,M[i].val);
printf("%lld",S);
return 0;
}