思路:ans[i]存储答案,枚举每一个数字a[i],对于a[i],先往左边扫,维护一个变量cnt和数组sum,
如果a[j]>a[i],cnt++,如果a[j]<a[i],cnt--,那么cnt表示的意思为大于a[i]的数字比小于a[i]的数字多多少个(为-表示少多少个)。每次sum[cnt]++,sum[cnt]的意思为(大于a[i]的数字比小于a[i]的数字多cnt个)的地方有几个。
完成以后我们再往右边扫描,注意从i位置开始,cnt=0,维护cnt,每次ans[i] += sum[-cnt]。意思为如果这一边大于a[i]的数字比小于a[i]的数字多cnt个,那么他与左边 (大于a[i]的数字比小于a[i]的数字少cnt个)的区间构成一个a[i]是中位数的区间。
这样n^2的复杂度就OK啦。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 8e3 + 5;
const int INF = 0x3f3f3f3f;
int n, a[MAXN], sum[MAXN*2], ans[MAXN];
int main()
{
while (~scanf("%d", &n))
{
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
for (int i = 0; i < n; i++)
{
memset(sum, 0, sizeof(sum));
int cnt = 0;
for (int j = i; j >= 0; j--)
{
if (a[j] > a[i]) cnt++;
if (a[j] < a[i]) cnt--;
sum[8000+cnt]++;
}
cnt = 0;
for (int j = i; j < n; j++)
{
if (a[j] > a[i]) cnt++;
if (a[j] < a[i]) cnt--;
ans[i] += sum[8000-cnt];
}
}
for (int i = 0; i < n; i++)
{
printf("%d", ans[i]);
if (i != n-1) printf(" ");
else printf("\n");
}
}
return 0;
}
/*
5
1 2 3 4 5
*/