中位数的中位数算法不一致

Question

我已经实现了选择/中位数算法的中位数使用以下作为参考http://www.ics.uci.edu/~eppstein/161/960130.html(这以前被链接在这里中位数在Java中)。

我的代码似乎适用于小数组(~100)，甚至适用于大小为100001 http://pastebin.com/mwRc4Hig(答案5008)的数组，但随后在大小为10001 http://pastebin.com/YwVBmgDk(答案4960，我的代码输出4958)的输入数组上失败。

请注意，上述文本的正确答案相当于对数组进行排序并返回array[array]中的元素。长度/2]，无论数组大小是偶数还是奇数。

我不知道如何调试这个问题。功能似乎是任意的，我只是迷路了。下面是我的代码:

public class MedianOfMedians {
public static void main(String[] args) {
    MedianOfMedians mds = new MedianOfMedians();
    mds.run();
}
private void run() {
    Scanner in = new Scanner(System.in);
    int n = in.nextInt();
    int[] numArray = new int[n];
    for (int i = 0; i < n; i++) {
        numArray[i] = in.nextInt();
    }
    int median = select(numArray, numArray.length / 2);
    System.out.print(median);
}
private int select(int[] numArray, int k) {
    if (numArray.length <= 10) {
        int[] sorted = insertionSort(numArray);
        return sorted[k];
    }
    int divCount = (numArray.length % 5 == 0) ? numArray.length / 5 - 1 : numArray.length / 5;
    int[] medOfMed = new int[divCount + 1];
    int counter = 0;
    int[] subArray;
    while (counter <= divCount) {
        subArray = splitByFive(counter, divCount, numArray);
        medOfMed[counter] = select(subArray, subArray.length / 2);
        counter++;
    }
    int M = select(medOfMed, numArray.length / 10);
    List<Integer> lt = new ArrayList<>();
    List<Integer> eq = new ArrayList<>();
    List<Integer> gt = new ArrayList<>();
    for (int i : numArray) {
        if (i < M) {
            lt.add(i);
        } else if (i == M) {
            eq.add(i);
        } else {
            gt.add(i);
        }
    }
    if (k < lt.size()) {
        return select(createArray(lt), k);
    } else if (k > lt.size() + eq.size()) {
        return select(createArray(gt), k - lt.size() - eq.size());
    } else {
        return M;
    }
}
private int[] splitByFive(int splitIter, int divisions, int[] toSplit) {
    int numToCopy;
    if (splitIter == divisions) {
        numToCopy = toSplit.length - (5 * splitIter);
    } else {
        numToCopy = 5;
    }
    int[] subArray = new int[numToCopy];
    System.arraycopy(toSplit, splitIter * 5, subArray, 0, numToCopy);
    return subArray;
}
private int[] createArray(List<Integer> list) {
    int[] result = new int[list.size()];
    for (int i = 0; i < list.size(); i++) {
        result[i] = list.get(i);
    }
    return result;
}
private int[] insertionSort(int[] numArray) {
    for (int i = 1; i < numArray.length; i++) {
        int j = i;
        while (j - 1 >= 0 && numArray[j] < numArray[j - 1]) {
            int temp = numArray[j];
            numArray[j] = numArray[j - 1];
            numArray[j - 1] = temp;
            j--;
        }
    }
    return numArray;
}
}

Answer 1

我没有时间调试你的代码，但也许我可以提供一种调试技术，让你自己尝试，这对像这样的递归算法很有用。

如果存在一个算法失败的输入(正如你所发现的那样)，那么就存在一个最小的这样的输入——这个输入越小，就越容易找出问题所在。因为算法是递归的，所以您有一个很好的方法来隔离第一个出错的地方:您可以在返回值之前测试即将从select()返回的结果是否正确(使用一种缓慢、可信的方法，例如将数据复制到临时缓冲区，对其进行排序，然后捕获中途元素)。如果您将函数重新排列为仅使用单个return语句，则这样做会容易得多，例如:

private int select(int[] numArray, int k) {
    int knownCorrectAnswer = selectSlowlyButDefinitelyCorrectly(numArray, k);
    int willReturn;
    if (numArray.length <= 10) {
        int[] sorted = insertionSort(numArray);
        willReturn = sorted[k];    // Just remember what we will return
    } else {    // Need to add else branch here now
        ...
        if (k < lt.size()) {
            willReturn = select(createArray(lt), k);
        } else if (k > lt.size() + eq.size()) {
            willReturn = select(createArray(gt), k - lt.size() - eq.size());
        } else {
            willReturn = M;
        }
    }    // End of inserted else branch
    if (willReturn == knownCorrectAnswer) {
        return willReturn;
    } else {
        yell("First problem occurs with numArray=<...> and k=<...>!");
    }
}

yell()应该打印出整个问题实例并停止程序(例如通过抛出异常)。这个设置的好处是，您知道当调用yell()时，对已经完成的select()的每个调用都是正确的—因为如果不是，yell()将已经被调用，并且程序将在此之前停止。因此，yell()产生的输出保证是出现问题的第一个子问题(不一定是最小的，但通常也是)。