HDU 5701中位数カウント暴力

2341 ワード

中位数
タイトル接続:
http://acm.hdu.edu.cn/showproblem.php?pid=5701
Description
中位数は、すべての値が小さい順から大きい順に真ん中に並ぶ数として定義され、値が偶数であれば、通常は真ん中の2つの数値の平均数を中位数としてとる.
現在n個の数があり,各数はユニークであり,各数がその区間を含むどれだけの中位数であるかを求める.
Input
複数組のテストデータ
第1行1個数n(n≦8000)
2行目のn個の数、0≦各数≦109、
Output
N個の数は,i番目の数がどれだけそれを含む区間で中位数であるかを順次示す.
Sample Input
5 1 2 3 4 5
Sample Output
1 2 3 2 1
Hint
に言及
問題:
暴力列挙中位数
そしてこの数より小さいのは-1、この数より大きいのは+1
sigma(num[x]*num[2 n-x])が彼のシナリオ数である限り
これはn^2暴力が無謀だったらいいのに
コード#コード#
#include<queue>
#include<map>
#include<stdio.h>
#include<cstring>
#include<set>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 8005;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n;
int a[maxn];
int H[maxn],mx;
int sum[maxn];
int l[maxn*2],r[maxn*2];
int tree[maxn];
vector<int>V;
int getid(int x)
{
    return lower_bound(V.begin(),V.end(),x)-V.begin()+1;
}
void solve()
{
    V.clear();
    for(int i=1;i<=n;i++)
        a[i]=read(),V.push_back(a[i]),H[i]=0;
    sort(V.begin(),V.end());
    V.erase(unique(V.begin(),V.end()),V.end());
    for(int i=1;i<=n;i++)a[i]=getid(a[i]);
    memset(H,0,sizeof(H));
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(a[j]<a[i])tree[j]=1;
            else if(a[j]==a[i])tree[j]=0;
            else tree[j]=-1;
        }
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        memset(sum,0,sizeof(sum));
        l[n]=1,r[n]=1;
        for(int j=i-1;j>=1;j--)
        {
            sum[j]=sum[j+1]+tree[j];
            l[sum[j]+n]++;
        }
        for(int j=i+1;j<=n;j++)
        {
            sum[j]=sum[j-1]+tree[j];
            r[sum[j]+n]++;
        }
        for(int j=0;j<=2*n-1;j++)
            H[i]+=l[j]*r[2*n-j];
    }
    printf("%d",H[1]);
    for(int i=2;i<=n;i++)
        printf(" %d",H[i]);
    printf("
"); } int main() { while(scanf("%d",&n)!=EOF) solve(); }