ZOJ 3499. Median

简介:     地址:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4322     题意:寻找中位数。对于一个(浮点数)数组,如果含有奇数个元素,“中位数”就是排序后位于数组中间那个。

    地址:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4322

    题意:寻找中位数。对于一个(浮点数)数组,如果含有奇数个元素,“中位数”就是排序后位于数组中间那个。如果含有偶数个元素,“中位数”是排序后数组中间那两个元素的平均值。

    

    此题应该是直接取自《算法导论》第9章(中位数和顺序统计学)的命题。如果对数组排序,再得出中位数,时间复杂度是 O(n log n) ,但这样的效率不高。对于这个命题不需要排序,算法导论的 9.2 节给出平均情况为 O(n)的方法,代码类似快排,其最差情况是 O(n^2),但由于对数组随机选取一个点作为分割点,所以最差情况几乎不可能发生。第 9.3 节给出一种最差情况也能达到线性时间的方法,但编码实现比较麻烦。书中也没有给出伪码。以下代码是第 9.2 节中的方法:

 

#include <stdio.h>
#include <stdlib.h>

double nums[500];

void exchange(double* A, int i, int j)
{
    double tmp;
    if(i != j)
    {
        tmp = A[i];
        A[i] = A[j];
        A[j] = tmp;
    }
}

/* 以A中最后一个元素作为分割点,把A就地划为为两个子序列 */
int fix_partition(double* A, int begin, int end)
{
    double x = A[end];
    int i = begin - 1, j;
    for(j = begin; j < end; j++)
    {
        if(A[j] <= x)
        {
            i++;
            exchange(A, i, j);
        }
    }
    exchange(A, i+1, end);
    return i + 1;
}

/* 从A中随即取一个元素作为分割点,把A就地划为为两个子序列 */
int random_partition(double* A, int begin, int end)
{
    int i = rand() % (end - begin + 1) + begin;
    exchange(A, i, end);
    return fix_partition(A, begin, end);
}

/* 返回数组 A[begin, ..., end] 中第 i 小的数字,即排序后的 A[begin + i -1]  */
double random_select(double* A, int begin, int end, int i)
{
    if(begin == end)
        return A[begin];

    int q = random_partition(A, begin, end);
    int k = q - begin + 1;
    
    if(i == k)
    {
        return A[q];
    }
    else if(i < k)
    {
        return random_select(A, begin, q-1, i);
    }
    else
    {
        return random_select(A, q+1, end, i-k);
    }
}


int main(int argc, char* argv[])
{
    int i, j;
    int T, n;
    scanf("%ld", &T);
    double result;
    for(i = 0; i < T; i++)
    {
        scanf("%ld", &n);
        for(j = 0; j < n; j++)
            scanf("%lf", nums + j);

        if(n & 1)
        {
            result = random_select(nums, 0, n-1, n/2+1);
        }
        else
        {
            result = random_select(nums, 0, n-1, n/2);
            result += random_select(nums, 0, n-1, n/2+1);
            result *= 0.5;
        }
        printf("%.3lf\n", result);
    }
    return 0;
}

 

    参考资料:

    (1)《算法导论》第9章,中位数和顺序统计学。

目录
相关文章
uva167 The Sultan's Successors
uva167 The Sultan's Successors
47 0
|
机器学习/深度学习
POJ 1775 (ZOJ 2358) Sum of Factorials
POJ 1775 (ZOJ 2358) Sum of Factorials
151 0
|
BI 人工智能
|
人工智能 机器学习/深度学习
POJ 1775 (ZOJ 2358) Sum of Factorials
Description John von Neumann, b. Dec. 28, 1903, d. Feb. 8, 1957, was a Hungarian-American mathematician who made important contributions t...
1152 0