【ZJOI2017 Round1练习】D2T3 counter（线段树）

题意：

思路：

预处理出b[i]代表i位置之前比a[i]小的数的个数

以每个数为结尾的组数是线段树中(1,a[i]-1)

对于a[i]换到最后，相当于线段树中(a[i]+1,n)--

交换后b[i]又变成了a[i]-1因为换到最后不需要考虑位置只需要考虑大小关系

 var t:array[..]of record
                            a,s,l:int64;
                           end;
     a,b:array[..]of longint;
     n,m,i:longint;
     ans,s:int64;

 procedure pushdown(p:longint);
 var l,r:longint;
     tmp:int64;
 begin
  l:=p<<; r:=l+;
  tmp:=t[p].a;
  if tmp= then exit;
  t[l].a:=t[l].a+tmp;
  t[r].a:=t[r].a+tmp;
  t[l].s:=t[l].s+t[l].l*tmp;
  t[r].s:=t[r].s+t[r].l*tmp;
  t[p].a:=;
 end;

 procedure pushup(p:longint);
 begin
  t[p].s:=t[p<<].s+t[p<<+].s;
 end;

 function query(l,r,x,y,p:longint):int64;
 var mid:longint;
 begin
  if x>y then exit();
  if (l>=x)and(r<=y) then exit(t[p].s);
  pushdown(p);
  mid:=(l+r)>>;
  query:=;
  if x<=mid then query:=query+query(l,mid,x,y,p<<);
  if y>mid then query:=query+query(mid+,r,x,y,p<<+);
  pushup(p);
 end;

 procedure update(l,r,x,y,v,p:longint);
 var mid:longint;
 begin
  if x>y then exit;
  if (l>=x)and(r<=y) then
  begin
   t[p].s:=t[p].s+t[p].l*v;
   t[p].a:=t[p].a+v;
   exit;
  end;
  pushdown(p);
  mid:=(l+r)>>;
  if x<=mid then update(l,mid,x,y,v,p<<);
  if y>mid then update(mid+,r,x,y,v,p<<+);
  pushup(p);
 end;

 procedure build(l,r,p:longint);
 var mid:longint;
 begin
  t[p].l:=r-l+;
  if l=r then
  begin
   t[p].s:=b[l];
   exit;
  end;
  mid:=(l+r)>>;
  build(l,mid,p<<);
  build(mid+,r,p<<+);
  pushup(p);
 end;

 procedure change(l,r,x,v,p:longint);
 var mid:longint;
 begin
  if l=r then
  begin
   t[p].a:=; t[p].s:=v;
   exit;
  end;
  pushdown(p);
  mid:=(l+r)>>;
  if x<=mid then change(l,mid,x,v,p<<)
   else change(mid+,r,x,v,p<<+);
  pushup(p);
 end;

 begin
  assign(input,'counter.in'); reset(input);
  assign(output,'counter.out'); rewrite(output);
  readln(n);
  build(,n,);
  for i:= to n do
  begin
   read(a[i]);
   b[a[i]]:=query(,n,,a[i]-,);
   update(,n,a[i],a[i],,);
  end;
  fillchar(t,sizeof(t),);
  build(,n,);
  for i:= to n do
  begin
   s:=query(,n,,a[i]-,);
   ans:=ans+s;
   update(,n,a[i]+,n,-,);
   s:=a[i]-;
   change(,n,a[i],s,);
  end;
  writeln(ans);

  close(input);
  close(output);
 end.