最小的k个数

题目描述

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

源代码：

解法一：没有通过

import java.util.ArrayList;
import java.util.Queue;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
     
    ArrayList<Integer> result = new ArrayList<Integer>();
       int len = input.length;
       if(k > len|| k == 0){
           return result;
       }
        int []tmp=new int[k];
        for(int i=0;i<k;i++)
            tmp[i]=input[i];
       
        for(int i=k;i<len;i++){
             int []maxarr = findmax(tmp);
             int max= maxarr[0];
             int maxind= maxarr[1];
            if(input[i]<max){
                updataatindex(tmp,maxind,input[i]);
            }
        }
      for(int i=0;i<k;i++)
          result.add(tmp[i]);
        return result;
    }
    public int[] findmax(int []arr){
        int []maxarr = new int[2];
        int len=arr.length;
        int max=arr[0];
        for(int i=1;i<len;i++){
            if(arr[i]>max){
                maxarr[0]=arr[i];
                maxarr[1]=i;
            } 
        }
        return maxarr;
    }
     public void updataatindex(int []arr,int index,int newval){
        arr[index]=newval;
    }
}

解法二：//通过了，但是没有看太懂

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(input.length < k){
            return list;
        }
        findKMin(input,0,input.length-1,k);
        for(int i = 0; i < k; i++){
            list.add(input[i]);
        }
        return list;
    }
    
public void findKMin(int[] a, int begin, int end,int k){
   if(begin < end){
   int pos = partition(a, begin, end);
   if(pos == k-1){
    return ;
   }else if(pos < k-1){
        findKMin(a,pos+1,end,k);
   }else{
       findKMin(a,begin,pos-1,k);
   }
  }
 }
 public int partition(int[] a, int begin, int end){
  int pivot = a[begin];
  while(begin < end){
   while(begin < end && a[end] >= pivot){end--;};
   a[begin] = a[end];
   while(begin < end && a[begin] <= pivot){begin++;};
   a[end] = a[begin];
  }
  a[begin] = pivot;
  return begin;
 }
}