福州大学第十五届程序设计竞赛题解

E - 谁还不是个宝宝

题目链接

http://acm.fzu.edu.cn/problem.php?pid=2288

题解

求奇数中位数区间，这种方法值得记录。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long LL;
int const maxn=1e3+10;
int a[maxn];
int cnt[maxn*2];

int main()
{
    int n;
    while(~scanf("%d",&n)){
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        for(int i=0;i<n;i++){
            memset(cnt,0,sizeof(cnt));
            int sum=0,sum2=0;
            int ans=1;
            for(int j=i-1;j>=0;j--){
                if(a[j]>a[i]) sum++;
                else sum2++;
                cnt[sum-sum2+1000]++;
                if(sum==sum2) ans++;
            }
            sum=sum2=0;
            for(int j=i+1;j<n;j++){
                if(a[j]>a[i]) sum++;
                else sum2++;
                if(sum==sum2) ans++;
                ans+=cnt[sum2-sum+1000];
            }
            if(i==n-1) printf("%d\n",ans);
            else printf("%d ",ans);
        }
    }
    return 0;
}