Find the Kth largest Element in the Array

http://www.geeksforgeeks.org/k-largestor-smallest-elements-in-an-array/

 

可以用冒泡排序。外循环k次，不用n次。O(nk)。

我想用quick sort的partition。每一次选择一个pivot，然后确定它的index。如果它的位置大于k，那么就在[left, index-1]找；如果它的位置小于k，那么就在[index+1, right]找。直到找到k。

这里不能用和quick sort完全一样的partition，因为那个partition不会返回pivot的index。

这里的做法是，先把pivot放到数组的末尾。然后对[left, right]按照pivot排序。最后跳出循环时，i一定对应一个大于等于pivot的值。这个时候再把pivot与A[i]交换，就能得到pivot的正确位置。

 

public static int findkth(int[] A, int k) {
        int left = 0, right = A.length-1;
        while (true) {
            int index = partition(A, left, right);
            if (index == A.length - k) return A[index];
            else if (index > A.length - k) right = index - 1;
            else left = index + 1;
        }
    }
    
    public static int partition(int[] A, int left, int right) {
        int i = left, j = right-1;
        int pivot = A[(left + right)/2];
        
        swap(A, (left + right) / 2, right);
        
        while (i <= j) {
            while (i < A.length && A[i] < pivot) {
                i++;
            }
            while (j >= 0 && A[j] > pivot) {
                j--;
            }
            if (i <= j) {
                swap(A, i, j);
                i++;
                j--;
            }
        }
        
        swap(A, i, right);
        
        return i;
    }
    
    public static void swap (int[] A, int i, int j) {
        int tmp = A[i];
        A[i] = A[j];
        A[j] = tmp;
    }