剑指offer-第五章优化时间和空间效率（最小的k个数）

题目：输入n个数，输出最小的k个数。

时间复杂度为O（n）

思路1：我们想的到的最直接的思路就是对这个N个数进行排序，然后就可以找到最小的k个了，同样可以用快排partition。但是只要找到前K个最小的元素，并不一定要排好序。

Java代码：

import java.util.Random;

//时间复杂度为O（n）；
public class PreKOfArray {
	//一趟快速排序
     public int partition(int[] a,int start,int end){
    	 if(a==null)
    		 return -1;
    	 Random rand=new Random();
         int index=start+rand.nextInt(end-start);
         swrap(a,index,end);
         int small=start-1;
         for(index=start;index<a.length;index++){
        	 if(a[index]<a[end]){
        		 small++;
        		 if(small!=index)
        			 swrap(a,index,small);
        	 }

         }
         small++;
         swrap(a,small,end);
         return small;
     }

	public  void swrap(int[] a, int index, int end) {
		int temp=a[index];
		a[index]=a[end];
		a[end]=temp;
	}
	public void printMinKOfArray(int[] input,int[] output,int k){
		if(input==null||output==null||k>input.length||k<=0)
			return;
		int start=0;
		int end=input.length-1;
		int index=partition(input,start,end);
		//找到前k个最小的数字
		while(index!=k-1){
			if(index>k-1){
				end=index-1;
				index=partition(input,start,end);
			}
			else{
				start=index+1;
				index=partition(input,start,end);
			}
		}

		for(int i=0;i<k;i++){
			output[i]=input[i];
		}

		for(int z:output)
			System.out.println(z+",");

	}
	public static void main(String[] args){
		int[] a={2,3,1,4,9,0,5,6,7,8};
		int k=5;
		int[] b=new int[k];
		PreKOfArray poa=new PreKOfArray();
		poa.printMinKOfArray(a, b, 5);
	}
}