POJ P2442 Sequence

题目大意:给你T组n*m的矩阵,每组矩阵每行取一个元素,组成一个包含n个元素的序列,一共有n^m种序列,让你求出序列和最小的前n个序列的序列和。

堆:
1.把第T组数据的第一组数组给记录。
2.从第二组到第N组开始,a[j]表示前i-1组第j大的序列和,i为当前第i组。
3.把前M大的a[j],跟这组数组组合,如果有一个比第M大的小就替换,然后替换堆并且做下移操作,因为做的是大头堆。
4.把最后求出来的f数组(堆)给排序一遍作为前i组数据前M大的序列和,加入到a数组中,做下一组数组。

var
   a,b,o,f:array [0..2001] of longint;
   t,i,j,k,l,n,m:longint;

procedure up(x:longint);
begin
   if x=1 then exit;
   repeat
     if f[x]>f[x div 2]
       then begin
               f[0]:=f[x];
               f[x]:=f[x div 2];
               f[x div 2]:=f[0];
            end
       else break;
     x:=x div 2;
  until x=0;
end;

procedure down(x:longint);
begin
  if 2*x>m then exit;
  repeat
    x:=2*x;
    if x+1<=m then
       if f[x]<f[x+1] then inc(x);
       if f[x div 2]<f[x] then
          begin
             f[0]:=f[x];
             f[x]:=f[x div 2];
             f[x div 2]:=f[0];
          end
        else break;
  until 2*x>m;
end;


procedure qsort(l,r:longint);
var
  i,j,mid:longint;
begin
  if l>=r then exit;
  i:=l; j:=r;
  mid:=o[(l+r) div 2];
  repeat
    while o[i]<mid do inc(i);
    while o[j]>mid do dec(j);
    if i<=j then
      begin
        o[0]:=o[i];
        o[i]:=o[j];
        o[j]:=o[0];
        inc(i); dec(j);
      end;
  until i>j;
  qsort(i,r);
  qsort(l,j);
end;

begin
    readln(t);
    for i:=1 to t do
        begin
            fillchar(a,sizeof(a),0);
            fillchar(b,sizeof(b),0);
            fillchar(f,sizeof(f),0);
            fillchar(o,sizeof(o),0);
            readln(n,m);
            for j:=1 to m do read(o[j]);
            qsort(1,m);
            a:=o;
            readln;
            for j:=1 to n-1 do
                begin
                   for k:=1 to m do read(o[k]);
                   readln;
                   qsort(1,m);
                   b:=o;
                   for k:=1 to m do
                       begin
                           f[k]:=a[k]+b[1];
                           up(k);
                       end;
                   for k:=2 to m do
                       for l:=1 to m do
                           if b[k]+a[l]<f[1]
                              then begin
                                      f[1]:=b[k]+a[l];
                                      down(1);
                                   end
                              else break;
                   o:=f;
                   qsort(1,m);
                   a:=o;
                   fillchar(f,sizeof(f),0);
            end;
          for j:=1 to m do write(a[j],' ');
          writeln;
        end;
end.
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值