【模板】最小生成树

题目描述
如题,给出一个无向图,求出最小生成树,如果该图不连通,则输出orz

输入输出格式
输入格式:
第一行包含两个整数N、M,表示该图共有N个结点和M条无向边。(N<=5000,M<=200000)
接下来M行每行包含三个整数Xi、Yi、Zi,表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式:
输出包含一个数,即最小生成树的各边的长度之和;如果该图不连通则输出orz

输入输出样例
输入样例#1:
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
输出样例#1:
7

说明
时空限制:1000ms,128M
数据规模:
对于20%的数据:N<=5,M<=20
对于40%的数据:N<=50,M<=2500
对于70%的数据:N<=500,M<=10000
对于100%的数据:N<=5000,M<=200000

样例解释:

这里写图片描述
所以最小生成树的总边权为2+2+3=7

var a,b,c,d:array[0..200000]of longint;
    i,j,n,ans,tot,x,m,k:longint;
function ask(x:longint):longint;
begin
  if d[x]=x then exit(d[x]);
  d[x]:=ask(d[x]);
  exit(d[x]);
end;
procedure unite(x,y:longint);
begin
  d[ask(x)]:=ask(y);
end;
procedure swap(var a,b:longint);
var t:longint;
begin
  t:=a; a:=b; b:=t;
end;
procedure sort(l,r:longint);
var
  i,j,x: longint;
begin
  i:=l; j:=r;
  x:=a[(l+r) div 2];
  repeat
  while a[i]<x do inc(i);
  while x<a[j] do dec(j);
  if not(i>j) then
  begin
    swap(a[i],a[j]);
    swap(b[i],b[j]);
    swap(c[i],c[j]);
    inc(i);
    j:=j-1;
  end;
  until i>j;
  if l<j then sort(l,j);
  if i<r then sort(i,r);
end;
begin
  read(n,m);
  for i:=1 to m do read(b[i],c[i],a[i]);
  sort(1,m);
  for i:=1 to n do d[i]:=i;
  while tot<n-1 do
  begin
    inc(k);
    if ask(b[k])<>ask(c[k])then
    begin
      inc(ans,a[k]);
      unite(b[k],c[k]);
      inc(tot);
    end;
  end;
  write(ans);
end.
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值