HDU 5701 中位数计数

#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn=8000+10;
int n;
int a[maxn],b[maxn];
int L[17000],R[17000];
vector<int>ans;

int main()
{
while(~scanf("%d",&n))
{
for(int i=1; i<=n; i++) scanf("%d",&a[i]);

ans.clear();
for(int i=1; i<=n; i++)
{
memset(b,0,sizeof b);
for(int j=1; j<=n; j++)
{
if(a[j]<a[i]) b[j]=-1;
else if(a[j]>a[i]) b[j]=1;
}

memset(L,0,sizeof L);
memset(R,0,sizeof R);

for(int j=i-1; j>=1; j--)
{
b[j]=b[j]+b[j+1];
L[b[j]+8000]++;
}

for(int j=i+1; j<=n; j++)
{
b[j]=b[j]+b[j-1];
R[b[j]+8000]++;
}

int tot=0;
for(int j=-8000; j<=8000; j++)
tot=tot+L[j+8000]*R[-j+8000];

tot=tot+L[0+8000];
tot=tot+R[0+8000];

ans.push_back(tot+1);
}
for(int i=0; i<ans.size(); i++)
{
printf("%d",ans[i]);
if(i<ans.size()-1) printf(" ");
else printf("\n");
}
}
return 0;
}