LeetCode 剑指offer 40.最小的k个数

题目描述

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

思路 

一上来就些思路 排序 取出前k个放在新数组里返回。

代码

import static java.util.Arrays.sort;
class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        sort(arr);
        int[] ret = new int[k];
        for (int i = 0; i <k ; i++) {
            ret[i] = arr[i];

        }
        return ret;
    }
}

题解

方法一 排序   

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int[] vec = new int[k];
        Arrays.sort(arr);
        for (int i = 0; i < k; ++i) {
            vec[i] = arr[i];
        }
        return vec;
    }
}
作者：LeetCode-Solution

思路基本一致 

方法二     堆，优先队列

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int[] vec = new int[k];//用于返回的数组
        if (k == 0) { // 排除 0 的情况
            return vec;
        }
        PriorityQueue<Integer> queue = new PriorityQueue<Integer>(new Comparator<Integer>() {//？重写conpare方法，规定比较标准呢
            public int compare(Integer num1, Integer num2) {
                return num2 - num1;//优先队列的比较标准:
            }
        });
        for (int i = 0; i < k; ++i) {
            queue.offer(arr[i]);//数组的前k个元素加入优先队列
        }
        for (int i = k; i < arr.length; ++i) {//让优先队列里的元素为最小的k个
            if (queue.peek() > arr[i]) {//peek查，如果队列元素比数组里元素大
                queue.poll();//poll取，删，取出队列元素
                queue.offer(arr[i]);//加入数组元素
            }
        }
        for (int i = 0; i < k; ++i) {
            vec[i] = queue.poll();
        }
        return vec;
    }
}

作者：LeetCode-Solution

反思总结

1 java.util.Arrays.sort; 方法 sort（数组）；默认升序排列

2 学习Java的优先队列操作

3 怎么把最小的k个加入队列没看懂 ， 后面在琢磨一下