【模板】最小生成树

来源:互联网 发布:手机视频短片软件 编辑:程序博客网 时间:2024/05/20 12:20

题目描述
如题,给出一个无向图,求出最小生成树,如果该图不连通,则输出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.
0 0
原创粉丝点击