p1627 [CQOI2009]中位数

传送门

分析

https://www.luogu.org/blog/user43145/solution-p1627

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define li long long
const int N = 1e5;
li a[*N+],b[*N+],Ans=;
int d[N+];
int main(){
    int n,m,i,j,k;
    scanf("%d%d",&n,&k);
    for(i=;i<=n;i++){
      scanf("%d",&d[i]);
      if(d[i]<k)d[i]=-;
        else if(d[i]>k)d[i]=;
        else d[i]=,j=i;
    }
    m=;
    for(i=j-;i>;i--){
      m+=d[i];
      a[m+N]++;
    }
    m=;
    for(i=j+;i<=n;i++){
      m+=d[i];
      b[m+N]++;
    }
    Ans+=a[N]+b[N];
    for(i=;i<=*N;i++)
      Ans+=a[i]*b[*N-i];
    cout<<Ans;
    return ;
}