中位数定义为所有值从小到大排序后排在正中间的那个数,如果值有偶数个,通常取最中间的两个数值的平均数作为中位数。
现在有n个数,每个数都是独一无二的,求出每个数在多少个包含其的区间中是中位数。
第一行一个数n(n<=8000) 第二行n个数,0<=每个数<=10^9
N个数,依次表示第i个数在多少包含其的区间中是中位数。
5 1 2 3 4 5
1 2 3 2 1
允许其他 AC 的用户查看此代码,分享代码才能查看别人的代码并有机会获得勋章
题解:一开始用了很烦的n^2的方法,A了这道题,后来发现利用前缀和便能很容易搞掉这道题,我们可以定义一个sum数组,用来统计前缀和,统计谁的前缀和呢,我们假设枚举到当前第i个数字,令比这个数大的数为1,比它小的数为-1,相等则为0,然后统计一发前缀和,前边的可以O(n)求出,后边可以O(1)查询,总复杂度为O(n^2)...
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<functional>
using namespace std;
typedef long long ll;
#define inf 1000000000
#define mod 1000000007
#define maxn 8050
#define lowbit(x) (x&-x)
#define eps 1e-9
int ans[maxn],sum[maxn],a[maxn],num[maxn*3];
int main(void)
{
int n,i,j,tmp;
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
for(i=1;i<=n;i++)
{
tmp=sum[0]=0;
memset(num,0,sizeof(num));
num[maxn]++;
for(j=1;j<=n;j++)
{
if(a[j]<a[i]) sum[j]=sum[j-1]-1;
else if(a[j]==a[i]) sum[j]=sum[j-1];
else sum[j]=sum[j-1]+1;
if(j<i) ++num[sum[j]+maxn];
else tmp+=num[sum[j]+maxn];
}
ans[i]=tmp;
}
for(i=1;i<=n;i++)
printf("%d ",ans[i]);
printf("\n");
return 0;
}