hdu 4908 BestCoder Sequence 策略

题意：

给定n和m，对于一个由1,2,3...,n组成的序列，问符合下列条件的子序列的个数：

1）子序列连续

2）子序列的中位数为m，

3）子序列长度为奇数

题解：

我们先找到m的位置，然后定义k=0，向左拓展，当遇到的数小于m时k++，否则k--，用f[0][i]存储各种k的出现次数（为了保证非负i=k+maxn）；同理向右拓展，当遇到的数小于m时k--，否则k++,用f[1][i]存储k出现的次数。那么结果就是ans=∑(f[0][i]*f[1][i])(maxn-n<=i<=maxn+n)

代码：

#include <cstdio>#include <cstring>#include <cmath>#include <cstdlib>#include <iostream>#include <algorithm>#include <vector>#include <map>#include <queue>#include <stack>using namespace std;const int maxn=4e4+10;int a[maxn],c[maxn],f[2][maxn*2];int main(){    //freopen("D:\\in.txt","r",stdin);    int n,m;    while(scanf("%d%d",&n,&m)!=EOF)    {        memset(f,0,sizeof(f));        int i,j,k=0,flag=0,x;        for(i=0;i<n;i++)        {            scanf("%d",&a[i]);            if(a[i]==m)x=i;        }        k=0;        i=x+1;        f[1][maxn]=f[0][maxn]=1;        for(;i<n;i++)        {            if(a[i]>m)k++;            else k--;            f[0][k+maxn]++;        }        i=x-1;        k=0;        for(;i>=0;i--)        {            if(a[i]>m)k--;            else k++;            f[1][k+maxn]++;        }        int ans=0;        for(i=-n;i<=n;i++)        {            ans+=f[0][i+maxn]*f[1][i+maxn];        }        printf("%d\n",ans);    }    return 0;}