CQOI2009 BZOJ1303 中位数

时间:2023-03-09 07:55:32
CQOI2009 BZOJ1303 中位数

首先找出b在数列中的位置mid

用 f[i]记录mid左边从mid往左统计比m小的数与比m大的数的差值为i的个数

用g[i]记录mid右边从mid往右统计比m大的数与比m小的数的差值为i的个数

。。有点语死早,找个样例模拟一下就懂了

 var n,b,mid,k,I:longint;
ans:int64;
a:array[..] of longint;
f,g:array[-..] of longint;
begin
readln(n,b);
for i:= to n do
begin
read(a[i]);
if b=a[i] then mid:=i;
end;
k:=;
for i:=mid downto do
begin
if a[i]>a[mid] then dec(k);
if a[i]<a[mid] then inc(k);
inc(f[k]);
end;
k:=;
for i:=mid to n do
begin
if a[i]>a[mid] then inc(k);
if a[i]<a[mid] then dec(k);
inc(g[k]);
end;
for i:=-n to n do ans:=ans+f[i]*g[i];
writeln(ans);
end.