随机选择算法

找到一个无序数组中第k大的元素，要求时间复杂度O（n）。

测试链接：. - 力扣（LeetCode）

public class RandomizedSelect {
    public static int findKthLargest(int[] nums, int k) {
        return randomizedSelect(nums, nums.length - k)
    }
    
    //如果arr排序的话，在i位置的数字是什么
    public static int randomizedSelect(int[] arr, int i) {
        int ans = 0;
        for (int l = 0, r = arr.length - 1; l <= r;) {
            partition(arr, l, r, arr[l + (int)(Math.random() * (r - l + 1))]);
            if (i < first) {
                r = first - 1;
            } else if (i > last) {
                l = last + 1;
            } else {
                ans = arr[i];
                break;
            }
        }
        return ans;
 
    }

    //荷兰国旗问题
    public static int first, last;

	public static void partition(int[] arr, int l, int r, int x) {
		first = l;
		last = r;
		int i = l;
		while (i <= last) {
			if (arr[i] == x) {
				i++;
			} else if (arr[i] < x) {
				swap(arr, first++, i++);
			} else {
				swap(arr, i, last--);
			}
		}
	}

	public static void swap(int[] arr, int i, int j) {
		int tmp = arr[i];
		arr[i] = arr[j];
		arr[j] = tmp;
	}
}

时间复杂度为什么是O（n）？

因为荷兰国旗问题的时间复杂度是O（n），在随机的情况下，第一次循环是在N的规模下产生一个荷兰国旗问题，第二次循环是在N/2的规模下产生的荷兰国旗问题，第三次循环是在N/4的规模下产生的荷兰国旗问题，因此时间复杂度为N + N/2 + N/4 + N/8 +...，因此时间复杂度为O（n）。