Company BDui 面试 笔试 : 从10个已按从大到小排列好的数组中，求top 15个最大的数

/* 
 * File:   bai  D company: 从10个已按从大到小排列好的数组中，求top 15个最大的数
 * Author: hongbin
 */


#include <cstdlib>
#include <iostream>
#include <vector>
#include<algorithm> 
using namespace std;


//array must be descending sorted when initialized. 
#define ARRAYNUM 10 //how many arrays whose values are sorteded descendingly
#define SIZE 10  //how many elements per an array
#define TOPNUM 15  //top 15 biggest numbers are listed


struct Current_Index{
int value;    
int ArrayIndex;
};


int data[ARRAYNUM][SIZE];


std::vector<int> HeadData; //save selected TOPNUM of values  


//element in an array  must be descending sorted, the first element is biggest. 
void InitData(){
 int i=0;
 int j=0;
    
 for(int i=0;i<ARRAYNUM;i++)
 for(int j=0;j<SIZE;j++) {
       data[i][j]=i*10-j;
    }
return;    
}


void export_topbig_data(){
  int index=0;
    int output_data;
    Current_Index myArrayIndex[ARRAYNUM];
    std::vector<int>::iterator myiter;
      //10 array index for 10 array, one to one
    //point to the first biggest element of each array
     for(int i=0;i<ARRAYNUM;i++) {
       myArrayIndex[i].ArrayIndex=0;
       myArrayIndex[i].value=data[i][0];
  //把每个队列的最前面的那个数字放到堆里面去。
       HeadData.push_back(data[i][0]);
    }
    //STL中关于heap默认调整成的是大顶堆: 即最前面的数值最大：
   make_heap(HeadData.begin(),HeadData.end());
   while( index < TOPNUM ) {
    //export current biggest data from the top of heap 
    output_data=HeadData.front();
    cout<<"Pop "<<index<<"th"<<" data: "<<output_data<<endl;
    /*
     input: 20 3 35 5 45
     make_heap后： 45 20 35 5 3 //the first is big one
     然后pop_heap： 35 20 3 5 45; //注意，pop_heap不是删除了45,而是把45挪到尾部来了，保证剩余的数 35 20 3 5是个大堆。   
     *  
     */
    pop_heap(HeadData.begin(),HeadData.end());  
    //remove this top data from vector 
    myiter=find(HeadData.begin(),HeadData.end(),output_data);
    
    if(myiter != HeadData.end())
    HeadData.erase(myiter);    
    //find which sub array includes this data
    for(int i=0;i<ARRAYNUM;i++ ) {
         if( myArrayIndex[i].value==output_data) {
           //push the next biggest data of this array into heap 
            myArrayIndex[i].ArrayIndex+=1;
            if( myArrayIndex[i].ArrayIndex < SIZE ) { 
              myArrayIndex[i].value=data[i][myArrayIndex[i].ArrayIndex];
              HeadData.push_back(myArrayIndex[i].value);
            }
            //re-make heap
            push_heap(HeadData.begin(),HeadData.end());            
            break;
        }   
      }
    index++;    
   }
   
   return;
}


int main(int argc, char** argv) {
       
    InitData();
    export_topbig_data();
  
  return 0;
}