[剑指offer] 最小的k个数 java

 一、堆排序

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res=new ArrayList<Integer>();
        if(k>input.length||input==null||k<=0) return res;
        for(int i=1;i<input.length;i++){//上溯过程，孩子比父亲小就交换
            while(input[i]<input[(i-1)/2]){
                swap(input,i,(i-1)/2);                
                i=(i-1)/2;
            }
        }
        res.add(input[0]);//这时候最小的肯定在堆顶，把它放到结果中
        int size=input.length-1;
        swap(input,0,size);堆顶已经添加过了，和最后一个数交换位置
        
        while(size>0&&k>1){
            heapify(input,0,size);//重新调整堆
            res.add(input[0]);//调整后堆顶依然最小，添加到结果中
            size--;//最后的数不要了，继续比较
            k--;//堆调整过一次，k就减1 直到k-1>0
            swap(input,0,size);
        }        
        return res;
    }
    public void heapify(int[] arr,int index,int size){
        int left=index*2+1;//左孩子
        while(left<size){
            int min=left+1<size&&arr[left+1]<arr[left]?left+1:left;//找左右孩子最小的
            min=arr[min]<arr[index]?min:index;//将自己与左右孩子中最小的一个比较大小
            if(min==index){   //自己最小，已经是小根堆，不需要调整            
                break;
            } 
           swap(arr,min,index); //自己不是最小，和孩子中最小的一个交换          
           left=index*2+1;     //继续向下走      
        }         
    }
    public void swap(int[] arr,int i,int j){
        int temp=arr[i];
        arr[i]=arr[j];
        arr[j]=temp;
    }
}

二、冒泡排序

 public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res=new ArrayList<Integer>();
        if(k<=0||k>input.length||input.length==0) return res;
        for(int i=0;i<k;i++){//一共走K趟
            for(int j=0;j<input.length-i-1;j++){
                if(input[j]<input[j+1]){//小数在前就交换
                    int temp=input[j];
                    input[j]=input[j+1];
                    input[j+1]=temp;
                }
            }
            res.add(input[input.length-i-1]);//把每趟排完的最后一个数加到结果中
        }
        return res;      
    }