题意

给你1-n的数，求以m为中位数的子序列有多少个

思路

l【i】表示m左边综合起来比m大的数个数为i

r【i】表示m左边综合起来比m小的数个数为i

那么在右面就可以找到综合起来比m小的个数为i的加上r【i】

比他大的加上l【i】

代码

/* **********************************************
Auther: 请叫我acm渣渣
Created Time: 2015-7-29 16:58:53
File Name   : hdu4908.cpp
*********************************************** */
#include <iostream>
#include <fstream>
#include <cstring>
#include <climits>
#include <deque>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
#include <utility>
#include <sstream>
#include <complex>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <ctime>
#include <bitset>
#include <functional>
#include <algorithm>
using namespace std;
#define lson L,m,rt<<1
#define rson m+1,R,rt<<1|1
#define ll long long
#define N 111111

int n,m;
int sum ;
int a[N];
int b[N];
int l[N],r[N];
int main(){
    while(scanf("%d%d",&n,&m)==2){
        int mid = 0;
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            if(a[i]==m) mid = i;
        }
        int cur =  0;
        for(int i=mid;i>0;i--){
            if(a[i]>m) cur++;
            if(a[i]<m) cur--;
            if(cur>0) l[cur]++;
            else r[-cur]++;
        }
        int ans= 0 ;
        cur = 0;
        for(int i=mid;i<=n;i++){
            if(a[i]>m) cur++;
            if(a[i]<m) cur--;
            if(cur>=0) ans+=r[cur];
            else ans+=l[-cur];
        }

        printf("%d\n",ans);
    }
}

版权声明：都是兄弟，请随意转载，请注明兄弟是谁