【枚举】【前缀和】【NOIP2011提高组Day1】选择客栈

 

【样例输入输出】

hotel.in	hotel.out
5 2 3 0 5 1 3 0 2 1 4 1 5	3

这一题朴素枚举能得60分，加上链表以及前缀和优化可以得100分！

由于只选两个客栈，所以很容易想到用枚举搜索的方法来实现，不过数据又太大，这就可以用前缀和（大家应该都懂吧，后面二分专题还会用到）的方式来进行优化，还需加上链表优化

话不多说，下面是Pascal代码，Pascal代码有点长，但是很详细

后面会给出C++ Code

Pascal Code

program hotel;

var
  n,kk,pp:longint;
  ans:longint;
  k,p:array[0..200000+10] of longint;
  num,sum,next:array[0..200000+10] of longint;

procedure init;
begin
  assign(input,'hotel.in');
  assign(output,'hotel.out');
  reset(input);
  rewrite(output);
end;

procedure outit;
begin
  close(input);
  close(output);
  halt;
end;

procedure readdata;
var
  i,j:longint;
begin
  read(n,kk,pp);
  
  for i:=1 to n do
  begin
    read(k[i],p[i]);
    sum[i]:=sum[i-1];
    if p[i]<=pp then inc(sum[i]);
  end;
  
  for i:=2 to n do
    for j:=i-1 downto 1 do
    begin
      if k[j]=k[i] then
      begin
        next[i]:=j;
        break;
      end;
    end;

  num[1]:=1;
  for i:=2 to n do
    num[i]:=num[next[i]]+1;
end;

procedure main;
var
  i,j,x:longint;
begin
  ans:=0;
  
  //朴素枚举+前缀和优化
  {for i:=1 to n do
  begin
    for j:=i+1 to n do//从i+1开始 避免重复计数
    begin
      if k[i]<>k[j] then continue;//色调不同 跳过
      if sum[j]-sum[i-1]>0 then inc(ans);
    end;
  end;}
  
  //链表枚举+前缀和优化
  for j:=2 to n do
  begin
    i:=next[j];
    while i<>0 do
    begin
      if sum[j]-sum[i-1]>0 then
      begin
        inc(ans,num[i]);
        break;
      end;
      i:=next[i];
    end;
  end;
  
  writeln(ans);
end;


begin
  init;
  readdata;
  main;
  outit;
end.