10.17T4 位置前后缀+map查询

2393 -- 【CQOI2009】中位数

Description

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

Input

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

Output

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

Sample Input

【输入1】

5 4

1 2 3 4 5

【输入2】

6 3

1 2 4 5 6 3

【输入3】

7 4

5 7 2 4 3 1 6

Sample Output

【输出1】

2

【输出2】

1

【输出3】

4

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

Hint

　　

 

 

 

 

 

在所有数里面，比m大的赋值1，比它小的赋值-1，然后找到这个m的位置，然后求出以它为最后一个元素的所有子段的和放进map里面维护

然后每次求出以这个元素为第一个元素的所有子段和，在map里面查找它的相反数个数加上就是答案，emmm算乱搞吧

code:

#include<iostream>
#include<cstdio>
#include<map>
using namespace std;
map<int,int>leftt,right;
int a[100005];
int main(){
    int n,m;
    cin>>n>>m;
    int pos;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]==m){
            pos=i;
            a[i]=0;
            continue;
        }
        a[i]=a[i]>m?1:-1;
    }
    int sum=0;
    for(int i=pos;i>=1;i--){
        sum+=a[i];
        leftt[sum]++;
    }
    sum=0;
    int ans=0;
    for(int i=pos;i<=n;i++){
        sum+=a[i];
        ans+=leftt[-sum];    
    }
    cout<<ans;
    return 0;
}

over