剑指offer——最小的K个数
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
利用堆排序中的构造小顶堆的方法,依次构造k次,然后就可以求出最小的k个数
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | import java.util.ArrayList; public class Solution { public ArrayList<Integer> GetLeastNumbers_Solution( int [] input, int k) { ArrayList<Integer> res = new ArrayList<>(); if (k > input.length) return res; int len = input.length; for ( int i = 0 ; i < k; i++){ smallHeap(input, len--); int temp = input[ 0 ]; input[ 0 ] = input[len]; input[len] = temp; res.add(input[len]); } return res; } public void smallHeap( int [] input, int high){ for ( int i = high/ 2 ; i > 0 ; i--){ int min; if ( 2 * i >= high) min = 2 * i; else { min = input[ 2 * i - 1 ] < input[ 2 * i] ? 2 * i : 2 * i + 1 ; } if (input[i - 1 ] > input[min - 1 ]) { int temp = input[i - 1 ]; input[i - 1 ] = input[min - 1 ]; input[min - 1 ] = temp; } } } } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步