1. 程式人生 > >HDU 5701 中位數計數 (暴力 思維題)

HDU 5701 中位數計數 (暴力 思維題)

中位數計數

Time Limit: 12000/6000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 354    Accepted Submission(s): 125

Problem Description 中位數定義為所有值從小到大排序後排在正中間的那個數,如果值有偶數個,通常取最中間的兩個數值的平均數作為中位數。
現在有n個數,每個數都是獨一無二的,求出每個數在多少個包含其的區間中是中位數。
Input 多組測試資料
第一行一個數n(n8000)
第二行n個數,0每個數109,
Output
N個數,依次表示第i個數在多少包含其的區間中是中位數。 Sample Input 5 1 2 3 4 5 Sample Output 1 2 3 2 1 Source  
題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=5701

題目分析:這題n方可過,顯然滿足條件的區間必是含奇數個數的區間,對於每個數,先往右掃一遍,求得其右邊比其大的和比其小的數的個數的差x,然後再往左掃一遍,求其左邊比起小的數和比起大的數的差,若一個數在這個區間為中位數,則若其右邊比它大的比比它小的多x(有點繞),則其左邊相反小的要比大的多x,這樣x才能正好在中間位置,用一個數組記錄一下差值為某個數的個數即可,注意要算上這個數自己。

#include <cstdio>
#include <cstring>
int const MAX = 8005;
int const CON = 8000;
int a[MAX], has[MAX << 1];

int main()
{
    int n;
    while(scanf("%d", &n) != EOF)
    {
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        for(int i = 1; i <= n; i++)
        {
            int cnt = 0, ans = 0;
            memset(has, 0, sizeof(has));
            has[CON] ++;
            for(int j = i + 1; j <= n; j++)
            {
                cnt += a[j] > a[i] ? 1 : -1;
                has[CON + cnt] ++;
            }
            cnt = 0;
            ans += has[CON];
            for(int j = i - 1; j >= 1; j--)
            {
                cnt += a[j] < a[i] ? 1 : -1;
                ans += has[CON + cnt];
            }
            printf("%d%c", ans, i == n? '\n' : ' ');
        }
    }
}