Pagini recente » Cod sursa (job #2074998) | Cod sursa (job #2335531) | Cod sursa (job #2489336) | Cod sursa (job #2950945) | Cod sursa (job #3258317)
#include <fstream>
#include <vector>
#define NMAX 200002
#define INF (1<<30)
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int N,M,H,S,nr,viz[NMAX],heap[NMAX],pozitie[NMAX],tata[NMAX];
vector <int>cost(NMAX,INF);
vector<vector<pair<int,int>>> A(NMAX);
struct solutie
{
int x,y;
} sol[NMAX];
void citire()
{
int x,y,c;
fin>>N>>M;
for(int i=1; i<=M; i++)
{
fin>>x>>y>>c;
A[x].push_back({y,c});
A[y].push_back({x,c});
}
}
void urcare(int poz)
{
while(poz/2>=1 && cost[heap[poz]]<cost[heap[poz/2]])
{
swap(pozitie[heap[poz]],pozitie[heap[poz/2]]);
swap(heap[poz],heap[poz/2]);
poz/=2;
}
}
void coborare(int poz)
{
while(2*poz<=H)
{
int r=2*poz;
if(r+1<=H && cost[heap[r+1]]<cost[heap[r]])
{
r++;
}
if(cost[heap[r]]<cost[heap[poz]])
{
swap(pozitie[heap[poz]],pozitie[heap[r]]);
swap(heap[poz],heap[r]);
poz=r;
}
else
{
break;
}
}
}
void adaugare_heap(int nod)
{
H++;
heap[H]=nod;
pozitie[nod]=H;
urcare(H);
}
void scoatere_minim()
{
swap(pozitie[heap[1]],pozitie[heap[H]]);
swap(heap[1],heap[H]);
H--;
coborare(1);
}
int main()
{
citire();
nr=0;
cost[1]=0;
adaugare_heap(1);
while(H)
{
int nod=heap[1];
scoatere_minim();
if(viz[nod])
{
continue;
}
S+=cost[nod];
viz[nod]=1;
if(tata[nod]!=0)
{
nr++;
sol[nr].x=tata[nod];
sol[nr].y=nod;
}
for(int j=0; j<A[nod].size(); j++)
{
if(!viz[A[nod][j].first] && cost[A[nod][j].first]>A[nod][j].second)
{
cost[A[nod][j].first]=A[nod][j].second;
tata[A[nod][j].first]=nod;
if(!pozitie[A[nod][j].first])
{
adaugare_heap(A[nod][j].first);
}
else
{
urcare(pozitie[A[nod][j].first]);
}
}
}
}
fout<< S << "\n";
fout<< nr << "\n";
for(int i=1; i<=nr; i++)
{
fout<< sol[i].x << " " << sol[i].y << "\n";
}
return 0;
}