解题6(OutputNMin)

题目描述

输入n个整数，输出其中最小的k个。

详细描述：

接口说明

原型：

bool GetMinK(unsignedint uiInputNum, int * pInputArray, unsignedint uiK, int * pOutputArray);

输入参数：

     unsignedint uiInputNum //输入整数个数

int * pInputArray  //输入整数数组

unsignedint uiK   //需输出uiK个整数

输出参数（指针指向的内存区域保证有效）：

    int * pOutputArray //最小的uiK个整数

返回值：

        false 异常失败

          true  输出成功

 

 

输入描述:

输入说明 
1 输入两个整数 
2 输入一个整数数组

输出描述:

输出一个整数数组

示例1

输入

5 2
1 3 5 7 2

输出

1 2

代码如下：

package com.yzh.hehe;

import java.util.Scanner;

public class OutputNMin {
/*
 * 先以o(n)时间建一个堆，再依次删除k个堆顶值
 *  
 * */ 
    public static void main(String[] args) { 
        Scanner scanner=new Scanner(System.in);
        while ( scanner.hasNext()) {
            int size=scanner.nextInt(); 
            int k=scanner.nextInt();
            int[]arr=new int[size+1]; 
            int temp;
            for (int i = 1; i <= size; i++) { 
                temp=scanner.nextInt();
                insertHeap(arr, temp,i);
            }
            for (int i = 0; i < k; i++) {
                deleteHeapMain(arr, size-i);
            } 
            int p=size-k+1; 
            for (int i =size ; i >p; i--) {
                System.out.print(arr[i]);
                System.out.println(" ");
            }
            System.out.print(arr[p]);
        }
        scanner.close();
        
    
         

    }
//建堆 执行n次建堆操作O(n)
     private static void insertHeap(int[] arr,int value,int p) {
         int i;
        for ( i =  p; i>1&&arr[i/2]>value; i=i/2) {
            arr[i]=arr[i/2];
        }
        arr[i]=value;  
    }
     //删除堆顶值（最小值） 
     private static void deleteHeapMain(int[]arr,int length) { 
            int reV=arr[1];//删除第一个值
            int lastV=arr[length];
            int i,child; 
            //2*i <=length保证空格还没下沉到最底层
            for (i = 1; 2*i <=length ;i=child ) {
                  child=2*i;
                  if(child!=length&&(arr[2*i]>arr[2*i+1])){
                      child++; //判断走左还是走右
                  }
                  //判断是否能将最后一个值插入到目前的空格
                  if(lastV>arr[child]){
                    arr[i]=arr[child];  
                  }else{
                      break;
                  }
            }
            arr[i]=lastV; //插入最后一个值
            arr[length]= reV;//用来存储删除的第一个值
        }
}