合并果子

题目大意

有n堆果子数目，需要合并成一堆。合并两堆果子需要的体力为两堆果子数目之和，求所需的最小体力值。

输入

3
1 2 9

输出

15

思路

此题时间复杂度O(nlog n)。

构建一棵小根堆，每次把堆顶和最后一个相加，重复操作，累加体力值，直到剩下一个值，堆操作请看上一篇。

代码

const
  maxn=100000;
type arr=array[1..maxn] of longint;
var
  f:arr;
  i,n,ans,sum:longint;
procedure heap(x,y:longint);
var
  s,i,j:longint;
begin
  i:=y;s:=f[y];j:=2*y;
  while j<=x do
    begin
      if (j<x)and(f[j]>f[j+1]) then inc(j);
      if s>f[j] then
        begin f[i]:=f[j];i:=j;j:=2*i;end
      else j:=x+1;
    end;
  f[i]:=s;
end;
begin
  readln(n);
  for i:=1 to n do read(f[i]);
  for i:=n div 2 downto 1 do
    heap(n,i);
  sum:=f[1];f[1]:=f[n];f[n]:=sum;
  heap(n-1,1);
  for i:=n-1 downto 1 do
    begin
      f[1]:=f[1]+f[i+1];
      ans:=ans+f[1];
      heap(i,1);
      sum:=f[1];f[1]:=f[i];f[i]:=sum;
      heap(i-1,1);
    end;
  writeln(ans)
end.