有效地（迭代）查找数组中第 k 个最小元素的索引？

Question

我想找到数组中第 k 个最小的元素，但实际上我的分区方法需要它的索引。

我在此博客上找到了用于查找第 k 个最小元素的代码： http://blog.teamleadnet.com/2012/07/quick-select-algorithm-find-kth-element.html

但这只会返回值，而不是索引。

您知道如何有效地找到它的索引吗？

Answer 1

最简单的方法是创建一个额外的相同长度的indices 数组，用从0 到length-1 的数字填充它，当arr 数组更改时，使用indices 执行相同的更改大批。最后从indices 数组中返回对应的条目。您甚至不必了解原始算法即可执行此操作。下面是修改后的方法（我的改动标有***）：

public static int selectKthIndex(int[] arr, int k) {
    if (arr == null || arr.length <= k)
        throw new IllegalArgumentException();

    int from = 0, to = arr.length - 1;

    // ***ADDED: create and fill indices array
    int[] indices = new int[arr.length];
    for (int i = 0; i < indices.length; i++)
        indices[i] = i;

    // if from == to we reached the kth element
    while (from < to) {
        int r = from, w = to;
        int mid = arr[(r + w) / 2];

        // stop if the reader and writer meets
        while (r < w) {

            if (arr[r] >= mid) { // put the large values at the end
                int tmp = arr[w];
                arr[w] = arr[r];
                arr[r] = tmp;
                // *** ADDED: here's the only place where arr is changed
                // change indices array in the same way
                tmp = indices[w];
                indices[w] = indices[r];
                indices[r] = tmp;
                w--;
            } else { // the value is smaller than the pivot, skip
                r++;
            }
        }

        // if we stepped up (r++) we need to step one down
        if (arr[r] > mid)
            r--;

        // the r pointer is on the end of the first k elements
        if (k <= r) {
            to = r;
        } else {
            from = r + 1;
        }
    }

    // *** CHANGED: return indices[k] instead of arr[k]
    return indices[k];
}

请注意，此方法会修改原始的 arr 数组。如果您不喜欢这样，请在方法的开头添加arr = arr.clone()。