[PTA]6-11 求自定类型元素序列的中位数

Spring-_-Bear 的 CSDN 博客导航

---

本题要求实现一个函数，求 N 个集合元素 A[] 的中位数，即序列中第 ⌊(N + 1) / 2⌋ 大的元素。其中集合元素的类型为自定义的 ElementType。

函数接口定义：

ElementType Median( ElementType A[], int N );

其中给定集合元素存放在数组 A[] 中，正整数 N 是数组元素个数。该函数须返回 N 个 A[] 元素的中位数，其值也必须是 ElementType 类型。

裁判测试程序样例：

#include <stdio.h>

#define MAXN 10
typedef float ElementType;

ElementType Median( ElementType A[], int N );

int main ()
{
    ElementType A[MAXN];
    int N, i;

    scanf("%d", &N);
    for ( i=0; i<N; i++ )
        scanf("%f", &A[i]);
    printf("%.2f\n", Median(A, N));

    return 0;
}

/* 你的代码将被嵌在这里 */

输入样例：

3
12.3 34 -5

输出样例：

12.30

来源：

来源：PTA | 程序设计类实验辅助教学平台
链接：https://pintia.cn/problem-sets/14/exam/problems/743

提交：

题解：

对数组元素进行排序，排序完成后直接返回第 A[N / 2] 个元素即可。亲测时间复杂度较高的冒泡排序、选择排序都要超时，二者时间复杂度均为 O(n²)，故选择平均时间复杂度较低的希尔排序算法 O(nlogn) 对数组元素进行排序。

希尔排序算法的基本思想：先将整个待排序的序列分割成为若干子序列，然后对子序列分别进行插入排序，待整个序列中的记录 “基本有序” 时，再对全体记录进行一次插入排序。

/*
 * 求 N 个集合元素 A[] 的中位数，即序列中第 (N + 1) / 2 大的元素
 */
ElementType Median(ElementType A[], int N) {
    // 按 gap 间隔对序列进行分组，直至 gap 为 1
    for (int gap = N; gap > 0; gap /= 2) {
        // 依次遍历每一组
        for (int i = gap; i < N; i++) {
            int cur = i;
            ElementType waitInsertVal = A[i];
            // 向前查找当前待插元素的插入位置（直至找到比当前元素小的元素）
            while (cur - gap >= 0 && A[cur - gap] - waitInsertVal > 0) {
                // 元素后移，为当前待插元素腾出空位
                A[cur] = A[cur - gap];
                cur -= gap;
            }
            A[cur] = waitInsertVal;
        }
    }

    return A[N / 2];
}