首页 > 试题广场 >

中位数

[编程题]中位数
  • 热度指数:5873 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 64M,其他语言128M
  • 算法知识视频讲解

M给你一个长度为n的数组,我们定义median数为该数组从小到大排序后,下标为(n-1)/2的数字。下标从0开始,(n-1)/2表示整数除法,即向下取整。现在我们已经得到了一个初始的数组,我们希望这个数组的median数是一个给定数字x。所以我们需要加入一些数到数组中从而完成我们的目标。数组中的元素可以重复,请问,最少需要加入多少个数字才能达成这个目标。


输入描述:
第一行输入两个整数n x (1 <= n <= 500, 1 <= x <= 10^5)。

接下来一行有n个正整数表示初始的数组,用空格分开,范围是[1, 10^5]。


输出描述:
输出需要加入最少的元素个数才能够使得新数组的median数成为x。
示例1

输入

3 2
2 3 4

输出

1

说明

样例一中,加入1,得到1 2 3 4,那么median数的下标为(4 - 1)/2 = 1, median数为2。加一个数字就可以了。
示例2

输入

3 4
1 2 3

输出

4

说明

样例二中,加入4 5 6 7,得到1 2 3 4 5 6 7,median数为4。最少加4个数字。
#include<stdio.h>

#define maxn 1000

int a[maxn];

// 快速排序函数
void quickSort(int l, int r) {
  if (l >= r) {   // 递归结束条件:区间只有一个元素或区间为空
    return;
  }
  int i = l, j = r, pivot = a[l]; // 定义左右指针和枢轴元素
  while (i < j) { // 左右指针相遇时退出循环
    while (i < j &&
           a[j] >= pivot) {    // 右指针向左移动,找到第一个小于枢轴元素的元素
      j--;
    }
    a[i] = a[j];    // 将该元素赋值给左指针所在位置
    while (i < j &&
           a[i] <= pivot) {    // 左指针向右移动,找到第一个大于枢轴元素的元素
      i++;
    }
    a[j] = a[i];    // 将该元素赋值给右指针所在位置
  }
  a[i] = pivot;   // 将枢轴元素放到最终正确的位置上
  quickSort(l, i - 1);    // 对左半部分区间递归进行快排
  quickSort(i + 1, r);    // 对右半部分区间递归进行快排
}

int main() {
  int n, x, i;
  scanf("%d%d", &n, &x);  // 输入数组长度和要插入的元素
  for (i = 0; i < n; i++) {
    scanf("%d", &a[i]); // 输入原始数组
  }

  quickSort(0, n - 1);    // 对原始数组进行排序

  int cnt = 0;    // 记录插入次数
  while (a[(n - 1) / 2] !=
         x) {   // 当中位数不等于要插入的元素时循环
    if (a[(n - 1) / 2] > x) {  // 如果中位数大于要插入的元素
      for (int k = n; k > (n + 1) / 2;
           k--) { // 将中位数及其右边的元素全部向右移动一位
        a[k] = a[k - 1];
      }
      a[(n + 1) / 2] = x;    // 将要插入的元素放到新的中间位置
    } else {    // 如果中位数小于要插入的元素
      for (int k = n; k >= (n + 1) / 2;
           k--) {    // 将中位数及其右边的元素全部向右移动一位
        a[k] = a[k - 1];
      }
      a[(n - 1) / 2] = x;    // 将要插入的元素放到新的中间位置
    }
    cnt++;  // 插入次数加1
    n++;    // 数组长度加1
    quickSort(0, n - 1);    // 对新的数组进行排序
  }

  printf("%d\n", cnt);

  return 0;
}

发表于 2023-11-21 16:25:01 回复(0)