剑指Offer--7.最小的K个数

题目描述:

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

示例 1：
输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]

示例 2：
输入：arr = [0,1,2,1], k = 1
输出：[0]

限制：
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

方法一： 直接使用快排

分析：
先用快速排序将整个数组排序，再取出前K个值即可。

代码：

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int[] res = new int[k];
        quickSort(arr, 0, arr.length-1);
        for(int i =0; i<k; i++){
            res[i] = arr[i];
        }
        return res;
    }

    public static void quickSort(int[] nums,int L , int R) {
		int left = L;
		int right = R;
		int temp = 0;
		if(right>left) {
			temp = nums[left];
			while(left!=right) {
				while(right>left&&nums[right]>=temp) {
					right--;  //右指针 直到找到比基准小的数					
				}
				nums[left] = nums[right]; // 将右指针的数 与 左指针的数调换
				while(right>left&&nums[left]<=temp) {
					left++;  // 左指针 直到找到比基准大的数				
				}
				nums[right] = nums[left];
			}
		nums[right] =temp;// 基准元素放回原来的位置
		quickSort(nums,L, left-1);
		quickSort(nums,right+1, R);
			
		}
    }
}

其他方法（快排/堆/二叉搜索树）

详见LeetCode