29、最小的K个数

一、题目

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

二、解法

 import java.util.ArrayList;
 public class Solution {
     public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
         // 运行时间：35ms 占用内存：503k
         ArrayList<Integer> minNs = new ArrayList<Integer>();
         if (k == 0 || k > input.length)
             return minNs;
         int[] m = new int[k];
         // 先把前k个存储到结果
         for (int j = 0; j < k; j++) {
             m[j] = input[j];
         }
         int maxIndex = 0;// 最大下标
         boolean findMax = true;// 默认每次都要找到最大的
         for (int i = k; i < input.length; i++) {
            if(findMax)
              maxIndex = 0;
             // 开始比较，从第k个开始，每次找最大的替换出来

                 for (int d = 1; d < k; d++) {
                     if (m[d] > m[maxIndex]) {
                         maxIndex = d;
                     }

             }
             // 找到后 比较 如果当前值input[k]比 m中的最大值还小，那么替换
             if (input[i] < m[maxIndex]) {
                 m[maxIndex] = input[i];
                 findMax = true;
             }else {
                 findMax = false;
             }
         }
         for (int i = 0; i < k; i++) {
             // 添加到minNs中
             minNs.add(m[i]);
         }
         return minNs;
     }
 }