BZOJ 1303: [CQOI2009]中位数图

BZOJ 1303: [CQOI2009]中位数图

题目

Time Limit: 1 Sec Memory Limit: 162 MB

Description

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

Input

第一行为两个正整数n和b ，第二行为1~n 的排列。

Output

输出一个整数，即中位数为b的连续子序列个数。

Sample Input

7 4

5 7 2 4 3 1 6

Sample Output

4

HINT

第三个样例解释：{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}

N<=100000

题解

首先，我们很容易想到一个暴力，先写一个相比较于数字b的前缀和，再枚举数字b(所在位置为p)左边的i个数字和右边的j个数字，使得sum[p]-sum[i-1]=sum[j]-sum[p-1]

很显然，用上面的这个方法，会超时

然后我就想到了hash，用hash记录下从位置p到位置1相比较于数字b的前缀和的值出现的次数，再枚举从位置p到位置n相比较于数字b的和t，ans：=ans+hash[t];

时间复杂度O（N）

代码（Pascal）

var n,m,ans:longint;
x:array[0..100005]of longint;
hash:array[-100005..100005]of longint;
i,t,p:longint;
begin
//assign(input,'1303.in');reset(input);
//assign(output,'1303.out');rewrite(output);
readln(n,m);t:=0;
for i:=1 to n do
begin
read(x[i]);
if x[i]=m then p:=i;
end;
inc(hash[0]);
for i:=p-1 downto 1 do
begin
if x[i]>m then inc(t) else dec(t);
inc(hash[t]);
end;
t:=0;
for i:=p to n do
begin
if x[i]>m then dec(t) else
if x[i]<m then inc(t);
ans:=ans+hash[t];
end;
write(ans);
//close(input);close(output);
end.