NOI2010超级钢琴 2

2006: [NOI2010]超级钢琴

Time Limit: 20 Sec  Memory Limit: 552 MB
Submit: 1296  Solved: 606
[Submit][Status]

Description

小Z是一个小有名气的钢琴家，最近C博士送给了小Z一架超级钢琴，小Z希望能够用这架钢琴创作出世界上最美妙的音乐。 这架超级钢琴可以弹奏出n个音符，编号为1至n。第i个音符的美妙度为Ai，其中Ai可正可负。 一个“超级和弦”由若干个编号连续的音符组成，包含的音符个数不少于L且不多于R。我们定义超级和弦的美妙度为其包含的所有音符的美妙度之和。两个超级和弦被认为是相同的，当且仅当这两个超级和弦所包含的音符集合是相同的。 小Z决定创作一首由k个超级和弦组成的乐曲，为了使得乐曲更加动听，小Z要求该乐曲由k个不同的超级和弦组成。我们定义一首乐曲的美妙度为其所包含的所有超级和弦的美妙度之和。小Z想知道他能够创作出来的乐曲美妙度最大值是多少。

Input

第一行包含四个正整数n, k, L, R。其中n为音符的个数，k为乐曲所包含的超级和弦个数，L和R分别是超级和弦所包含音符个数的下限和上限。 接下来n行，每行包含一个整数Ai，表示按编号从小到大每个音符的美妙度。

Output

只有一个整数，表示乐曲美妙度的最大值。

Sample Input

4 3 2 3

3

2

-6

8

Sample Output

11

【样例说明】
共有5种不同的超级和弦：

音符1 ~ 2，美妙度为3 + 2 = 5
音符2 ~ 3，美妙度为2 + (-6) = -4
音符3 ~ 4，美妙度为(-6) + 8 = 2
音符1 ~ 3，美妙度为3 + 2 + (-6) = -1
音符2 ~ 4，美妙度为2 + (-6) + 8 = 4
最优方案为：乐曲由和弦1，和弦3，和弦5组成，美妙度为5 + 2 + 4 = 11。

HINT

 

Source

Day1

题解：

唉呀，真是太感动了，竟然这么快就改好了，我还以为要一天呢，不能多说，激动啊

代码：（写得很丑）

const maxn=600000+10;inf=maxlongint;
type node=record
     x,y,z:longint;
     end;
var s,t:array[0..21,0..maxn] of longint;
    a,b:array[0..maxn] of longint;
    i,j,n,m,x,y,k,l,r,tmp:longint;
    ans:int64;
    c:array[0..2*maxn] of node;
    w:node;
  function max(x,y:longint):longint;
   begin
   if x>y then exit(x) else exit(y);
   end;

  procedure modify(x,y,z:longint);
   var k,i:longint;
   begin
   c[0].x:=x;c[0].y:=y;c[0].z:=z;
   i:=1;k:=2;
   while k<=m do
    begin
      if (k<m) and (c[k+1].x>c[k].x) then inc(k);
      if c[0].x<c[k].x then begin c[i]:=c[k];i:=k;k:=k<<1;end
      else k:=m+1;
    end;
   c[i]:=c[0];
   end;
 procedure insert(x,y,z:longint);
   var k,i:longint;
   begin
   inc(m);c[0].x:=x;c[0].y:=y;c[0].z:=z;
   i:=m;k:=i>>1;
   while k>=1 do
     begin
      if c[0].x>c[k].x then begin c[i]:=c[k];i:=k;k:=i>>1;end
      else k:=0;
     end;
   c[i]:=c[0];
   end;
procedure sort(l,r:longint);
 var i,j,m,temp:longint;
 begin
 i:=l;j:=r;x:=a[(i+j)>>1];
 repeat
  while b[a[i]]<b[x] do inc(i);
  while b[a[j]]>b[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 i<r then sort(i,r);
 if j>l then sort(l,j);
 end;
procedure build(h,l,r:longint);
 var i,p,mid:longint;
 begin
 mid:=(l+r)>>1;p:=0;
 for i:=l to r do
  if t[h,i]<=mid then
   begin
    t[h+1,l+p]:=t[h,i];
    inc(p);
    s[h,i]:=p;
   end
  else
   begin
    t[h+1,mid+1+i-p-l]:=t[h,i];
    s[h,i]:=p;
   end;
 if l=r then exit;
 build(h+1,l,mid);
 build(h+1,mid+1,r);
 end;
function find(h,l,r,x,y,k:longint):longint;
 var ll,rr,mid:longint;
 begin
 if l=r then exit(t[h,l]);
 mid:=(l+r)>>1;
 if l=x then ll:=0 else ll:=s[h,x-1];rr:=s[h,y]-ll;
 if rr>=k then exit(find(h+1,l,mid,l+ll,l+rr+ll-1,k))
 else exit(find(h+1,mid+1,r,mid+1+x-l-ll,mid+1+y-l-rr-ll,k-rr));
 end;
procedure init;
  begin
    readln(n,k,l,r);inc(n);
    for i:=2 to n do begin readln(x);b[i]:=b[i-1]+x;end;
    for i:=1 to n do a[i]:=i;
    sort(1,n);
  end;
procedure main;
   begin
   for i:=1 to n do t[0,a[i]]:=i;
   build(0,1,n);//writeln(n);
   for i:=l+1 to n do
     begin
      tmp:=b[a[find(0,1,n,max(i-r,1),i-l,1)]];
    //  writeln(tmp);
      insert(b[i]-tmp,i,1);
     end;
   ans:=0;
   for i:=1 to k do
     begin
      w:=c[1]; // writeln(w.x);
      inc(ans,w.x);
      if (w.z>=r-l+1) or ((w.y<r+1) and (w.z>w.y-l-1)) then modify(-inf,inf,inf)
      else
       begin
        tmp:=b[w.y]-b[a[find(0,1,n,max(w.y-r,1),w.y-l,w.z+1)]];
        modify(tmp,w.y,w.z+1);
       end;
     end;
   writeln(ans);
   end;
begin
 assign(input,'input.txt');assign(output,'output.txt');
 reset(input);rewrite(output);
 init;
 main;
 close(input);close(output);
end.