HDU 4908 BestCoder Sequence，hdu4908

HDU 4908 BestCoder Sequence，hdu4908

將大於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;}