首页 > 代码库 > HDU 4908 BestCoder Sequence

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;
}