剑指offer-最小的k个数

Posted 2022-01-07 zhacai

题目描述

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

 

最下topk，使用大顶堆

    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) 
        ArrayList<Integer> re = new ArrayList<Integer>();
        if(input == null || input.length<k ||k==0)//判断非法
            return re;
        
        int[] heap = new int[k];//new一个堆空间
        for(int i=0;i<k;i++)
            heap[i] = input[i];
        
        for(int i = k/2;i>=0;i--)//调整大顶堆
            adjustHeap(heap,i,k-1);
        
        for(int i = k;i<input.length;i++)//对第k个以后的数，如果小于堆顶元素就和堆顶替换，否则不进行任何操作
            if(input[i]<heap[0])
                heap[0] = input[i];
                adjustHeap(heap,0,k-1);
            
        
        for(int i = 0;i<k;i++)
            re.add(heap[i]);
        
        return re;
    
    public void adjustHeap(int[] heap,int left ,int right)//调整大顶堆
        int target = heap[left];
        for(int i=left*2+1;i<=right;i=i*2+1)
            if(i<right&&heap[i]<heap[i+1])
                i++;
            
            if(heap[i]<=target)
                break;
            
            heap[left] = heap[i];
            left =i;
        
        heap[left] =target;