BZOJ 1303: [CQOI2009]中位数图

1303: [CQOI2009]中位数图

Time Limit: 1 Sec  Memory Limit: 162 MB
[Submit][Status][Discuss]

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

Source

 题目链接

 

因为是n的排列，b只会出现一次，b记为0，比b小记为-1，比b大记为1，所求为和为0的序列个数

然后从值为b的点向前向后分别统计一下前缀和，互为相反数的组数就是答案

C++数组下标不能是负的感觉这题写Pascal更方便一点QAQ

 

var
    n, k, x, q, ans, i: longint;
    a : array [1 .. 1010000] of integer;
    s : array [-1010000 .. 1010000] of longint;
begin
    readln(n, k);
    for i := 1 to n do
    begin
        read(x);
        if (x = k) 
        then begin
            a[i] := 0;
            q := i;
        end
        else begin
            if (x > k) then a[i] := 1 else a[i] := -1;
        end;
    end;
    x := 0;
    ans := 1;
    for i := q - 1 downto 1 do
    begin
        x := x + a[i];
        s[x] := s[x] + 1;
        if (x = 0) then ans := ans + 1;
    end;
    x := 0;
    for i := q + 1 to n do
    begin
        x := x + a[i];
        ans := ans + s[-x];
        if (x = 0) then ans := ans + 1;
    end;
    writeln(ans);
end.