BZOJ1303: [CQOI2009]中位数图

C++程序计数问题解析

转载于 2017-12-18 23:40:00 发布 · 61 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/w-h-h/p/8060363.html

本文介绍了一个使用C++解决特定计数问题的方法。该程序通过读取一系列整数并找出与给定值相等的整数的位置，然后计算前后两部分中小于和大于给定值的元素数量之差的组合数。此算法适用于竞赛编程和问题求解。

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<vector>
#define MAXN 100000+10
#define a(x) a[x+MAXN]
#define b(x) b[x+MAXN]
using namespace std;
int a[MAXN<<1],b[MAXN<<1];
int n,m,p;
int s[MAXN];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
        if(s[i]==m){
            p=i;
        }    
    }
    int fe=0,bi=0;
    a(0)++;
    for(int i=p-1;i>=1;i--){
        if(s[i]<m)fe++;
        else bi++;
        a(fe-bi)++;
    }
    fe=0,bi=0;
    b(0)++;
    for(int i=p+1;i<=n;i++){
        if(s[i]<m)fe++;
        else bi++;
        b(fe-bi)++;
    }
    int ans=0;
    for(int i=-n;i<=n;i++){
        ans+=a(i)*b(-i);
    }
    printf("%d\n",ans);
    return 0;
}