題目描述

輸入n個整數，找出其中最小的K個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,。

解題思路

解法1
大頂堆，存放k個數字，遍歷數組，當堆的數量小于k，則直接加入堆，否則與堆頂元素比較，如果比堆頂元素小，則將堆頂元素移除，并添加當前元素。

解法2
我們可以根據快速排序的思想，找到數組中第k個元素，排在其前面的都是比他小的。

實現

• 解法1

import java.util.ArrayList; import java.util.TreeSet;public class Solution {public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {ArrayList<Integer> list = new ArrayList<>();if (input == null || input.length < k || k <= 0) return list;TreeSet<Integer> set = new TreeSet<>();for (int i : input) {if (set.size() < k) set.add(i);else {if (set.last() > i) {set.pollLast();set.add(i);}}}list.addAll(set);return list;} }

• 解法2

import java.util.ArrayList; import java.util.Random; public class Solution {public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {ArrayList<Integer> list = new ArrayList<>();if (input == null || input.length < k || k <= 0) return list;//TreeSet<Integer> set = new TreeSet<>();int par = partion(input, 0, input.length-1);int start = 0, end = input.length - 1;while (par != k-1){if (par < k - 1){start = par + 1;par = partion(input,par + 1, end);}else {end = par - 1;par = partion(input,start,end-1);}}for (int i = 0; i < k; i++){list.add(input[i]);}return list;}private int partion(int[] input, int start, int end) {if (start > end) return 0;Random r = new Random();int rd = r.nextInt(end- start + 1) + start;int tmp = input[rd];input[rd] = input[end];input[end] = tmp;int index =start -1;for (int i = start; i < end; i++){if (input[i] <= input[end]){index++;if (index != i){tmp = input[index];input[index] = input[i];input[i] = tmp;}}}tmp = input[++index];input[index] = input[end];input[end] = tmp;return index;} }

轉載于:https://www.cnblogs.com/ggmfengyangdi/p/5782646.html

總結

以上是生活随笔為你收集整理的剑指Offer_29_最小的K个数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。