Медиана медиан выбрать алгоритм

Question

последние несколько дней я очень старался написать этот алгоритм, но безуспешно. Код работает, и в большинстве случаев он дает мне правильный результат, но в некоторых случаях он терпит неудачу. Например, с этим массивом {3, 8, 1, 9, 10, 7, 6, 2, 5, 4} и k = 6 это должно дать мне 6 как результат, но это даст мне 7. Может ли кто-нибудь мне помочь? Я не могу понять, в чем проблема.

Вот код:

class MOMSelect {

static int median(int a[], int i, int n) {
    if(i <= n)
        Arrays.sort(a, i, n);
    else
        Arrays.sort(a, n, i);
    return a[n/2];
}

static int medianOfMediansSelect(int a[], int left, int right, int k) {
    int n = right - left + 1;
    int i;
    int[] medians = new int[(n + 4) / 5];
    for (i = 0; i < n/5; i++) {
        medians[i] = median(a, left + i * 5, 5);
    }
    if (i*5 < n) {
        medians[i] = median(a,left + i * 5, n % 5);
        i++;
    }
    int medianOfMedians = (i == 1)? medians[i - 1]: median(medians, 0, medians.length);
    int pivotIndex = partition(a, left, right, medianOfMedians);
    if (pivotIndex == k - 1) {
        return a[pivotIndex];
    }
    else if (pivotIndex - left > k - 1) {
        return medianOfMediansSelect(a, left, pivotIndex - 1, k);
    }
    else {
        return medianOfMediansSelect(a, pivotIndex + 1, right, k);
    }
}

static void swap(int[] a, int i, int j) {
    int temp = a[i];
    a[i] = a[j];
    a[j] = temp;
}

static int partition(int[] a, int left, int right, int x) {
    int i;
    for (i = left; i < right; i++)
        if (a[i] == x)
            break;
    swap(a, i, right);
    i = left;
    for (int j = left; j <= right - 1; j++) {
        if(a[j] <= x) {
            swap(a, i, j);
            i++;
        }
    }
    swap(a, i, right);
    return i;
}

public static void main(String[] args) {
    int a[] = {3, 8, 1, 9, 10, 7, 6, 2, 5, 4};
    int n = a.length;
    int k = 1;
    System.out.println(medianOfMediansSelect(a, 0, n - 1, k));
}
}

Заранее всем спасибо

Answer 1

Хорошо, я решил. Помимо моего плохого понимания метода Arrays.sort (), была глупая ошибка в структуре if, где я проверял значение pivotPosition для метода medianOfMediansSelect()

Точнее в этой строке

else if (pivotIndex - left > k - 1) {

Я должен был сделать так

else if (pivotIndex > k - 1) {

Answer 2

Итак, я изменил ваш метод median, чтобы исправить проблему. Вы должны проверить Arrays.sort метод для ясного понимания.

Теперь он дает 6, для k=6.

Вот оно,

   static int median(int a[], int i, int n)
   {
      Arrays.sort(a, i, i + n - 1);

      return a[i + n / 2];
   }