【堆】B-Station

B-Station

试题描述

在离著名的国家Berland不远的地方，有一个水下工作站。这个工作站有N层。已知：是第i层装有W_i的水，最多可以容纳L_i的水，恐怖分子炸毁第i层的代价是P_i。第i层一旦被炸毁，该层所有的水都将倾泻到第i+1层。如果某一层的水量超过了它的容量(即L_i)，那么该层就将自动被毁坏，所有的水也会倾泻到下一层。

Pivland的恐怖分子想要用最少的钱毁掉第N层，现在他雇佣你来计算，需要炸毁哪些层。

输入

第一行有一个自然数N(1<=n<=15000)。接下来的N行，每行3个整数Wi, Li, Pi(0<=Wi,Li,Pi<=15000)。

输出

输出需要炸毁的层的编号。

样例

Input

3

1000 1000 1

0 1000 2

2 10 100

 

Output

1

2

=====================

=================

朴素

-------------------

type  node=record         w,l,p:longint;       end;var  n:longint;  ti:array[0..150000]of node;procedure init;begin  assign(input,'bstation.in');  assign(output,'bstation.out');  reset(input); rewrite(output);end;procedure terminate;begin  close(input); close(output);  halt;end;procedure main;var  i,j:longint;  ans,w,k:longint;  s:longint;begin  readln(n);  for i:=1 to n do    readln(ti[i].w,ti[i].l,ti[i].p);  ans:=ti[n].p; s:=n;  for i:=n-1 downto 1 do  //枚举从i开始炸    begin      k:=ti[i].p;      w:=ti[i].w;      if k>ans then continue;      for j:=i+1 to n do        begin          w:=w+ti[j].w;;          if w<=ti[j].l then            begin              k:=k+ti[j].p;              if k>ans then break;            end;        end;      if k<ans then        begin          ans:=k;          s:=i;        end;    end;  w:=ti[s].w;  writeln(s);  for i:=s+1 to n do    begin      w:=w+ti[i].w;      if w<=ti[i].l then writeln(i);    end;end;begin  init;  main;  terminate;end.

 

_________________________________

堆优化

---------------------------

var  n:longint;  sum,w,l,p:array[0..150000]of longint;  dui:array[0..150000]of longint;  dui_s:longint;  procedure init;begin  assign(input,'bstation.in');  assign(output,'bstation.out');  reset(input); rewrite(output);end;procedure terminate;begin  close(input); close(output);  halt;end;procedure shift(r,n:longint);var  k:longint;  tem:longint;begin  k:=2*r;  if (k+1<=n) and (sum[dui[k+1]]-l[dui[k+1]]>sum[dui[k]]-l[dui[k]])    then inc(k); //保证大根堆..  while (k<=n)and(sum[dui[k]]-l[dui[k]]>sum[dui[r]]-l[dui[r]]) do    begin      tem:=dui[k];      dui[k]:=dui[r];      dui[r]:=tem;      r:=k;      k:=2*r;      if (k+1<=n) and (sum[dui[k+1]]-l[dui[k+1]]>sum[dui[k]]-l[dui[k]])        then inc(k); //保证大根堆..    end;end;procedure adjust(r:longint);var  k:longint;  tem:longint;begin  k:=r shr 1;  while ((k>0) and (sum[dui[k]]-l[dui[k]]<sum[dui[r]]-l[dui[r]])) do  //保证上面的比较大..    begin      tem:=dui[k];      dui[k]:=dui[r];      dui[r]:=tem;      r:=k;      k:=r shr 1;    end;end;procedure main;var  i,j:longint;  ans:longint;  t:longint;  s:longint;  water:longint;begin  readln(n);  sum[0]:=0;  for i:=1 to n do    begin      readln(w[i],l[i],p[i]);      sum[i]:=sum[i-1]+w[i];    end;  ans:=p[n];  dui_s:=0;  inc(dui_s);  dui[dui_s]:=n;  //入堆  t:=ans;  for i:=n-1 downto 1 do    //表示从i开始炸.    begin      t:=t+p[i];  //记录变化{由上一次的状态而来}..      inc(dui_s);      dui[dui_s]:=i;      adjust(dui_s); //调整堆..      while sum[dui[1]]-l[dui[1]]>sum[i-1] do      //当不用炸的时侯        begin          t:=t-p[dui[1]];  //减去那个不需要炸得价值..          dui[1]:=dui[dui_s];          dec(dui_s);          shift(1,dui_s);        end;      if t<ans then        begin          ans:=t;          s:=i;        end;    end;  water:=w[s]; writeln(s);  for i:=s+1 to n do    begin      water:=water+w[i];      if water<=l[i] then writeln(i);    end;end;begin  init;  main;  terminate;end.