Question

所以我实现了＃34;找到数组中的第k个最小元素＆＃34;使用改进的快速排序算法。但是，现在它是无限循环。我不太清楚错误在哪里。 更新：调试器说错误在第14行：＆＃34;返回kthSmallestError（arr，start，j-1，k）;根据print语句，（start，j-1，k）值为（3,3,0）＆＃34; 感谢您的帮助！

class kthSmallestElement {
    public static void main(String[] args) {
        int[] input = {3, 1, 5, 2, 6, 4, 7};
        int result = kthSmallestElement(input, 0, input.length-1, 3);
        System.out.println(result);
    }
    public static int kthSmallestElement(int[] arr, int start, int end, int k) {
        int j = partition(arr, start, end);
        if (j == k) return arr[j];
        if (j < k) {
            return kthSmallestElement(arr, j+1, end, k-j-1);
        }
        else {
            return kthSmallestElement(arr, start, j-1, k);
        }
    }

    public static int partition(int[] arr, int left, int right) {
        int pivot = arr[left+(right-left)/2];

        while (left <= right) {
            while (arr[left] < pivot) {
                left++;
            }
            while (arr[right] > pivot) {
                right--;
            }
            if (left <= right) {
                int temp = arr[left];
                arr[left] = arr[right];
                arr[right] = temp;
                left++;
                right--;
            }
        }
        return left;
    }
}

Answer 1

一个明显的错误：left, right, j是k相对的绝对指数。 j和k之间的所有比较和算术都应该由start校准。

Answer 2

请试试这个。

感谢，

- 约翰

    public static int kthSmallestElement(int[] arr, int start, int end, int k) {
    //int j = partition(arr, start, end);
    int j = partition2(arr, start, end);
    if (j == k)
        return arr[j];
    if (j < k) {
        // without -1 as original code.
        return kthSmallestElement(arr, j + 1, end, k - j );
    } else {
        return kthSmallestElement(arr, start, j - 1, k);
    }
}

public static int partition2(int[] arr, int lo, int hi) {
    int left = lo;
    int right = hi+1;
    int p = arr[lo];

    while (true) {
        while (arr[++left] < p) {
            if (left == hi)
                break;
        }

        while (p < arr[--right]) {
            if (right == lo) {
                break;
            }
        }

        if (left >= right) {
            break;
        }

        exchange(arr, left, right);
    }

    exchange(arr, lo, right);
    return right;
}

使用Quicksort无限循环错误的Java kthSmallestElement

2 个答案: