hdu 5701/百度之星初赛试题 中位数计数 思维

中位数定义为所有值从小到大排序后排在正中间的那个数，如果值有偶数个，通常取最中间的两个数值的平均数作为中位数。

现在有n个数，每个数都是独一无二的，求出每个数在多少个包含其的区间中是中位数。

Input

多组测试数据

第一行一个数n(n≤8000)

第二行nn个数，0≤每个数≤109

Output

N个数，依次表示第i个数在多少包含其的区间中是中位数。

Sample Input

5

1 2 3 4 5

Sample Output

1 2 3 2 1

思路:

考虑一下一个数怎么样才能在中间,即这个数左面比他大的数与比他小的数差为x,那么其右边比他小的数与比他大的数的差一定也为x.

对于这个题数据量这么小,n方可以过.那么我们就开一个数组先扫一下左边记录比他大的数与比他小的数的差,在扫右面比他小的数与比他大的数的差,记录一下解即可.

需要特别注意的情况就是当这个数的左边和右边x为0的情况.

#include<bits/stdc++.h>
#define Ri(a) scanf("%d", &a)
#define Rl(a) scanf("%lld", &a)
#define Rf(a) scanf("%lf", &a)
#define Rs(a) scanf("%s", a)
#define Pi(a) printf("%d\n", (a))
#define Pf(a) printf("%lf\n", (a))
#define Pl(a) printf("%lld\n", (a))
#define Ps(a) printf("%s\n", (a))
#define W(a) while(a--)
#define CLR(a, b) memset(a, (b), sizeof(a))
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define exp 0.00000001
#define  pii  pair<int, int>
#define  mp   make_pair
#define  pb   push_back
using namespace std;
typedef long long ll;
const int maxn=1e4+10;
ll gcd(ll a,ll b)
{
return b==0?a:gcd(b,a%b);
}
int a[maxn],b[2*maxn];
int n;
int main()
{
while(~Ri(n))
{
for(int i=1;i<=n;i++)
Ri(a[i]);
ll ans=0,cnt=0;
for(int i=1;i<=n;i++)
{
ans=0;
cnt=0;
CLR(b,0);
b[8000]++;
for(int j=i-1;j>=1;j--)
{
cnt+=a[j]>a[i]  ?  1 : -1;
b[cnt+8000]++;
}
ans+=b[8000];
cnt=0;
for(int j=i+1;j<=n;j++)
{
cnt+=a[j]<a[i] ? 1 : -1;
ans+=b[cnt+8000];
}
printf("%d%c", ans, i == n? '\n' : ' ');
}
}
return 0;
}