算法复习——堆排

Long long ago，我曾经想过要写堆排（话说已经是刚开始写二分那个博客的时候了），后来好久没写，现在总算想到要写了……

好了，进入正题。

我以前曾经自己打过一个课件，现在打算放在CSDN下载那里，就当是放基础知识了，现在写写一道题的题解……

PPT网址如下（下载者先交1积分……）：http://download.csdn.net/detail/u014120397/9472391

题目：codevs1245

链接：http://codevs.cn/problem/1245/

K路归并，初始化的时候把a,b数组排个序，然后把a[1]+b[1]，a[2]+b[1]……a[n]+b[1]放入小根堆里面，然后每出一次堆，比如说把a[i]+b[1]出堆了，那么就把a[i]+b[2]入堆，如此操作执行n次就可以了。时间复杂度O(n log n)

代码：

type arr=array[1..100000]of longint;     ijk=record     sum,b:longint;     end;var heap:array[0..1000000]of ijk;    buf:ijk;    a,b:arr;    n,i,length:longint;procedure qs(l,r:longint;var a:arr);var i,j:longint;    x,y:longint;begin  i:=l;  j:=r;  x:=a[(l+r) shr 1];  repeat    while a[i]<x do    inc(i);    while a[j]>x do    dec(j);    if i<=j then    begin      y:=a[i];      a[i]:=a[j];      a[j]:=y;      inc(i);      dec(j);    end;  until i>j;  if l<j then qs(l,j,a);  if i<r then qs(i,r,a);end;procedure minheapify(x:longint);var min,l,r:longint;begin  l:=x shl 1;  r:=l+1;  min:=x;  if (l<=length)and(heap[l].sum<heap[min].sum) then  min:=l;  if (r<=length)and(heap[r].sum<heap[min].sum) then  min:=r;  if min<>x then  begin    heap[0]:=heap[x];    heap[x]:=heap[min];    heap[min]:=heap[0];    minheapify(min);  end;end;procedure sinkup(i:longint);var parent,tmp:longint;begin  parent:=i shr 1;  if (parent<>0)and(heap[parent].sum>heap[i].sum) then  begin    heap[0]:=heap[i];    heap[i]:=heap[parent];    heap[parent]:=heap[0];    if parent>1 then sinkup(parent);  end;end;function popheap:ijk;var min:ijk;begin  if length=0 then exit;  min:=heap[1];  heap[1]:=heap[length];  dec(length);  minheapify(1);  exit(min);end;procedure pushheap(buf:ijk);begin  inc(length);  heap[length]:=buf;  if length>1 then  sinkup(length);end;begin  readln(n);  for i:=1 to n do  read(a[i]);  for i:=1 to n do  read(b[i]);  qs(1,n,a);  qs(1,n,b);  for i:=1 to n do  begin    buf.sum:=a[i]+b[1];    buf.b:=1;    pushheap(buf);  end;  for i:=1 to n do  begin    buf:=popheap;    if i<>n then write(buf.sum,' ')else writeln(buf.sum);    if buf.b<=n then    begin      buf.sum:=buf.sum-b[buf.b]+b[buf.b+1];      inc(buf.b);      pushheap(buf);    end;  end;end.