HDU 4908 BestCoder Sequence

將大於M的數標記爲1 小於M的數標記爲-1 M本身標記爲0 

則題目就是要求和爲0並且包括M的連續序列的個數

用sum[i]表示從第1個數大牌第i個數的標記的和

對於所有大於等於M的位置的i 我們要求小於M的位置的sum[j]==sum[i]的個數的和即爲答案

#include<bits.stdc++.h>
using namespace std;
int sum[40020],f[80020];
int main()
{
    int N,M,k,a;
    while(scanf("%d%d",&N,&M)==2)
    {
        memset(f,0,sizeof(f));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&a);
            if(a>M)
                sum[i]=sum[i-1]+1;
            else if(a<M)
                sum[i]=sum[i-1]-1;
            else
            {
                k=i;
                sum[i]=sum[i-1];
            }
        }
        for(int i=0;i<=k-1;i++)
            f[sum[i]+40000]++;
        int ans=0;
        for(int i=k;i<=N;i++)
            ans+=f[sum[i]+40000];
        printf("%d\n",ans);
    }
    return 0;
}